gusucode.com > 各种VC自绘控件源码程序 > 各种VC自绘控件源码/code/SkinControls(自绘MFC基本控件 )/SkinControls/SkinControls/CxImage/ximadsp.cpp
// xImaDsp.cpp : DSP functions
/* 07/08/2001 v1.00 - Davide Pizzolato - www.xdp.it
* CxImage version 6.0.0 02/Feb/2008
*/
#include "ximage.h"
#include "ximaiter.h"
#if CXIMAGE_SUPPORT_DSP
////////////////////////////////////////////////////////////////////////////////
/**
* Converts the image to B&W.
* The OptimalThreshold() function can be used for calculating the optimal threshold.
* \param level: the lightness threshold.
* \return true if everything is ok
*/
bool CxImage::Threshold(BYTE level)
{
if (!pDib) return false;
if (head.biBitCount == 1) return true;
GrayScale();
CxImage tmp(head.biWidth,head.biHeight,1);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
for (long y=0;y<head.biHeight;y++){
info.nProgress = (long)(100*y/head.biHeight);
if (info.nEscape) break;
for (long x=0;x<head.biWidth;x++){
if (BlindGetPixelIndex(x,y)>level)
tmp.BlindSetPixelIndex(x,y,1);
else
tmp.BlindSetPixelIndex(x,y,0);
}
}
tmp.SetPaletteColor(0,0,0,0);
tmp.SetPaletteColor(1,255,255,255);
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Converts the image to B&W, using a threshold mask
* \param pThresholdMask: the lightness threshold mask.
* the pThresholdMask image must be grayscale with same with and height of the current image
* \return true if everything is ok
*/
bool CxImage::Threshold(CxImage* pThresholdMask)
{
if (!pDib) return false;
if (head.biBitCount == 1) return true;
if (!pThresholdMask) return false;
if (!pThresholdMask->IsValid() ||
!pThresholdMask->IsGrayScale() ||
pThresholdMask->GetWidth() != GetWidth() ||
pThresholdMask->GetHeight() != GetHeight()){
strcpy(info.szLastError,"invalid ThresholdMask");
return false;
}
GrayScale();
CxImage tmp(head.biWidth,head.biHeight,1);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
for (long y=0;y<head.biHeight;y++){
info.nProgress = (long)(100*y/head.biHeight);
if (info.nEscape) break;
for (long x=0;x<head.biWidth;x++){
if (BlindGetPixelIndex(x,y)>pThresholdMask->BlindGetPixelIndex(x,y))
tmp.BlindSetPixelIndex(x,y,1);
else
tmp.BlindSetPixelIndex(x,y,0);
}
}
tmp.SetPaletteColor(0,0,0,0);
tmp.SetPaletteColor(1,255,255,255);
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Filters only the pixels with a lightness less (or more) than the threshold level,
* and preserves the colors for the unfiltered pixels.
* \param level = the lightness threshold.
* \param bDirection = false: filter dark pixels, true: filter light pixels
* \param nBkgndColor = filtered pixels are set to nBkgndColor color
* \param bSetAlpha = if true, sets also the alpha component for the filtered pixels, with nBkgndColor.rgbReserved
* \return true if everything is ok
* \author [DP], [wangsongtao]
*/
////////////////////////////////////////////////////////////////////////////////
bool CxImage::Threshold2(BYTE level, bool bDirection, RGBQUAD nBkgndColor, bool bSetAlpha)
{
if (!pDib) return false;
if (head.biBitCount == 1) return true;
CxImage tmp(*this, true, false, false);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
tmp.GrayScale();
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*y/head.biHeight);
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
BYTE i = tmp.BlindGetPixelIndex(x,y);
if (!bDirection && i<level) BlindSetPixelColor(x,y,nBkgndColor,bSetAlpha);
if (bDirection && i>=level) BlindSetPixelColor(x,y,nBkgndColor,bSetAlpha);
}
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract RGB channels from the image. Each channel is an 8 bit grayscale image.
* \param r,g,b: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitRGB(CxImage* r,CxImage* g,CxImage* b)
{
if (!pDib) return false;
if (r==NULL && g==NULL && b==NULL) return false;
CxImage tmpr(head.biWidth,head.biHeight,8);
CxImage tmpg(head.biWidth,head.biHeight,8);
CxImage tmpb(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long y=0; y<head.biHeight; y++){
for(long x=0; x<head.biWidth; x++){
color = BlindGetPixelColor(x,y);
if (r) tmpr.BlindSetPixelIndex(x,y,color.rgbRed);
if (g) tmpg.BlindSetPixelIndex(x,y,color.rgbGreen);
if (b) tmpb.BlindSetPixelIndex(x,y,color.rgbBlue);
}
}
if (r) tmpr.SetGrayPalette();
if (g) tmpg.SetGrayPalette();
if (b) tmpb.SetGrayPalette();
/*for(long j=0; j<256; j++){
BYTE i=(BYTE)j;
if (r) tmpr.SetPaletteColor(i,i,0,0);
if (g) tmpg.SetPaletteColor(i,0,i,0);
if (b) tmpb.SetPaletteColor(i,0,0,i);
}*/
if (r) r->Transfer(tmpr);
if (g) g->Transfer(tmpg);
if (b) b->Transfer(tmpb);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract CMYK channels from the image. Each channel is an 8 bit grayscale image.
* \param c,m,y,k: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitCMYK(CxImage* c,CxImage* m,CxImage* y,CxImage* k)
{
if (!pDib) return false;
if (c==NULL && m==NULL && y==NULL && k==NULL) return false;
CxImage tmpc(head.biWidth,head.biHeight,8);
CxImage tmpm(head.biWidth,head.biHeight,8);
CxImage tmpy(head.biWidth,head.biHeight,8);
CxImage tmpk(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long yy=0; yy<head.biHeight; yy++){
for(long xx=0; xx<head.biWidth; xx++){
color = BlindGetPixelColor(xx,yy);
if (c) tmpc.BlindSetPixelIndex(xx,yy,(BYTE)(255-color.rgbRed));
if (m) tmpm.BlindSetPixelIndex(xx,yy,(BYTE)(255-color.rgbGreen));
if (y) tmpy.BlindSetPixelIndex(xx,yy,(BYTE)(255-color.rgbBlue));
if (k) tmpk.BlindSetPixelIndex(xx,yy,(BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue));
}
}
if (c) tmpc.SetGrayPalette();
if (m) tmpm.SetGrayPalette();
if (y) tmpy.SetGrayPalette();
if (k) tmpk.SetGrayPalette();
if (c) c->Transfer(tmpc);
if (m) m->Transfer(tmpm);
if (y) y->Transfer(tmpy);
if (k) k->Transfer(tmpk);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract YUV channels from the image. Each channel is an 8 bit grayscale image.
* \param y,u,v: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitYUV(CxImage* y,CxImage* u,CxImage* v)
{
if (!pDib) return false;
if (y==NULL && u==NULL && v==NULL) return false;
CxImage tmpy(head.biWidth,head.biHeight,8);
CxImage tmpu(head.biWidth,head.biHeight,8);
CxImage tmpv(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long yy=0; yy<head.biHeight; yy++){
for(long x=0; x<head.biWidth; x++){
color = RGBtoYUV(BlindGetPixelColor(x,yy));
if (y) tmpy.BlindSetPixelIndex(x,yy,color.rgbRed);
if (u) tmpu.BlindSetPixelIndex(x,yy,color.rgbGreen);
if (v) tmpv.BlindSetPixelIndex(x,yy,color.rgbBlue);
}
}
if (y) tmpy.SetGrayPalette();
if (u) tmpu.SetGrayPalette();
if (v) tmpv.SetGrayPalette();
if (y) y->Transfer(tmpy);
if (u) u->Transfer(tmpu);
if (v) v->Transfer(tmpv);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract YIQ channels from the image. Each channel is an 8 bit grayscale image.
* \param y,i,q: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitYIQ(CxImage* y,CxImage* i,CxImage* q)
{
if (!pDib) return false;
if (y==NULL && i==NULL && q==NULL) return false;
CxImage tmpy(head.biWidth,head.biHeight,8);
CxImage tmpi(head.biWidth,head.biHeight,8);
CxImage tmpq(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long yy=0; yy<head.biHeight; yy++){
for(long x=0; x<head.biWidth; x++){
color = RGBtoYIQ(BlindGetPixelColor(x,yy));
if (y) tmpy.BlindSetPixelIndex(x,yy,color.rgbRed);
if (i) tmpi.BlindSetPixelIndex(x,yy,color.rgbGreen);
if (q) tmpq.BlindSetPixelIndex(x,yy,color.rgbBlue);
}
}
if (y) tmpy.SetGrayPalette();
if (i) tmpi.SetGrayPalette();
if (q) tmpq.SetGrayPalette();
if (y) y->Transfer(tmpy);
if (i) i->Transfer(tmpi);
if (q) q->Transfer(tmpq);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract XYZ channels from the image. Each channel is an 8 bit grayscale image.
* \param x,y,z: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitXYZ(CxImage* x,CxImage* y,CxImage* z)
{
if (!pDib) return false;
if (x==NULL && y==NULL && z==NULL) return false;
CxImage tmpx(head.biWidth,head.biHeight,8);
CxImage tmpy(head.biWidth,head.biHeight,8);
CxImage tmpz(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long yy=0; yy<head.biHeight; yy++){
for(long xx=0; xx<head.biWidth; xx++){
color = RGBtoXYZ(BlindGetPixelColor(xx,yy));
if (x) tmpx.BlindSetPixelIndex(xx,yy,color.rgbRed);
if (y) tmpy.BlindSetPixelIndex(xx,yy,color.rgbGreen);
if (z) tmpz.BlindSetPixelIndex(xx,yy,color.rgbBlue);
}
}
if (x) tmpx.SetGrayPalette();
if (y) tmpy.SetGrayPalette();
if (z) tmpz.SetGrayPalette();
if (x) x->Transfer(tmpx);
if (y) y->Transfer(tmpy);
if (z) z->Transfer(tmpz);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Extract HSL channels from the image. Each channel is an 8 bit grayscale image.
* \param h,s,l: pointers to CxImage objects, to store the splited channels
* \return true if everything is ok
*/
bool CxImage::SplitHSL(CxImage* h,CxImage* s,CxImage* l)
{
if (!pDib) return false;
if (h==NULL && s==NULL && l==NULL) return false;
CxImage tmph(head.biWidth,head.biHeight,8);
CxImage tmps(head.biWidth,head.biHeight,8);
CxImage tmpl(head.biWidth,head.biHeight,8);
RGBQUAD color;
for(long y=0; y<head.biHeight; y++){
for(long x=0; x<head.biWidth; x++){
color = RGBtoHSL(BlindGetPixelColor(x,y));
if (h) tmph.BlindSetPixelIndex(x,y,color.rgbRed);
if (s) tmps.BlindSetPixelIndex(x,y,color.rgbGreen);
if (l) tmpl.BlindSetPixelIndex(x,y,color.rgbBlue);
}
}
if (h) tmph.SetGrayPalette();
if (s) tmps.SetGrayPalette();
if (l) tmpl.SetGrayPalette();
/* pseudo-color generator for hue channel (visual debug)
if (h) for(long j=0; j<256; j++){
BYTE i=(BYTE)j;
RGBQUAD hsl={120,240,i,0};
tmph.SetPaletteColor(i,HSLtoRGB(hsl));
}*/
if (h) h->Transfer(tmph);
if (s) s->Transfer(tmps);
if (l) l->Transfer(tmpl);
return true;
}
////////////////////////////////////////////////////////////////////////////////
#define HSLMAX 255 /* H,L, and S vary over 0-HSLMAX */
#define RGBMAX 255 /* R,G, and B vary over 0-RGBMAX */
/* HSLMAX BEST IF DIVISIBLE BY 6 */
/* RGBMAX, HSLMAX must each fit in a BYTE. */
/* Hue is undefined if Saturation is 0 (grey-scale) */
/* This value determines where the Hue scrollbar is */
/* initially set for achromatic colors */
#define HSLUNDEFINED (HSLMAX*2/3)
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::RGBtoHSL(RGBQUAD lRGBColor)
{
BYTE R,G,B; /* input RGB values */
BYTE H,L,S; /* output HSL values */
BYTE cMax,cMin; /* max and min RGB values */
WORD Rdelta,Gdelta,Bdelta; /* intermediate value: % of spread from max*/
R = lRGBColor.rgbRed; /* get R, G, and B out of DWORD */
G = lRGBColor.rgbGreen;
B = lRGBColor.rgbBlue;
cMax = max( max(R,G), B); /* calculate lightness */
cMin = min( min(R,G), B);
L = (BYTE)((((cMax+cMin)*HSLMAX)+RGBMAX)/(2*RGBMAX));
if (cMax==cMin){ /* r=g=b --> achromatic case */
S = 0; /* saturation */
H = HSLUNDEFINED; /* hue */
} else { /* chromatic case */
if (L <= (HSLMAX/2)) /* saturation */
S = (BYTE)((((cMax-cMin)*HSLMAX)+((cMax+cMin)/2))/(cMax+cMin));
else
S = (BYTE)((((cMax-cMin)*HSLMAX)+((2*RGBMAX-cMax-cMin)/2))/(2*RGBMAX-cMax-cMin));
/* hue */
Rdelta = (WORD)((((cMax-R)*(HSLMAX/6)) + ((cMax-cMin)/2) ) / (cMax-cMin));
Gdelta = (WORD)((((cMax-G)*(HSLMAX/6)) + ((cMax-cMin)/2) ) / (cMax-cMin));
Bdelta = (WORD)((((cMax-B)*(HSLMAX/6)) + ((cMax-cMin)/2) ) / (cMax-cMin));
if (R == cMax)
H = (BYTE)(Bdelta - Gdelta);
else if (G == cMax)
H = (BYTE)((HSLMAX/3) + Rdelta - Bdelta);
else /* B == cMax */
H = (BYTE)(((2*HSLMAX)/3) + Gdelta - Rdelta);
// if (H < 0) H += HSLMAX; //always false
if (H > HSLMAX) H -= HSLMAX;
}
RGBQUAD hsl={L,S,H,0};
return hsl;
}
////////////////////////////////////////////////////////////////////////////////
float CxImage::HueToRGB(float n1,float n2, float hue)
{
//<F. Livraghi> fixed implementation for HSL2RGB routine
float rValue;
if (hue > 360)
hue = hue - 360;
else if (hue < 0)
hue = hue + 360;
if (hue < 60)
rValue = n1 + (n2-n1)*hue/60.0f;
else if (hue < 180)
rValue = n2;
else if (hue < 240)
rValue = n1+(n2-n1)*(240-hue)/60;
else
rValue = n1;
return rValue;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::HSLtoRGB(COLORREF cHSLColor)
{
return HSLtoRGB(RGBtoRGBQUAD(cHSLColor));
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::HSLtoRGB(RGBQUAD lHSLColor)
{
//<F. Livraghi> fixed implementation for HSL2RGB routine
float h,s,l;
float m1,m2;
BYTE r,g,b;
h = (float)lHSLColor.rgbRed * 360.0f/255.0f;
s = (float)lHSLColor.rgbGreen/255.0f;
l = (float)lHSLColor.rgbBlue/255.0f;
if (l <= 0.5) m2 = l * (1+s);
else m2 = l + s - l*s;
m1 = 2 * l - m2;
if (s == 0) {
r=g=b=(BYTE)(l*255.0f);
} else {
r = (BYTE)(HueToRGB(m1,m2,h+120) * 255.0f);
g = (BYTE)(HueToRGB(m1,m2,h) * 255.0f);
b = (BYTE)(HueToRGB(m1,m2,h-120) * 255.0f);
}
RGBQUAD rgb = {b,g,r,0};
return rgb;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::YUVtoRGB(RGBQUAD lYUVColor)
{
int U,V,R,G,B;
float Y = lYUVColor.rgbRed;
U = lYUVColor.rgbGreen - 128;
V = lYUVColor.rgbBlue - 128;
// R = (int)(1.164 * Y + 2.018 * U);
// G = (int)(1.164 * Y - 0.813 * V - 0.391 * U);
// B = (int)(1.164 * Y + 1.596 * V);
R = (int)( Y + 1.403f * V);
G = (int)( Y - 0.344f * U - 0.714f * V);
B = (int)( Y + 1.770f * U);
R= min(255,max(0,R));
G= min(255,max(0,G));
B= min(255,max(0,B));
RGBQUAD rgb={(BYTE)B,(BYTE)G,(BYTE)R,0};
return rgb;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::RGBtoYUV(RGBQUAD lRGBColor)
{
int Y,U,V,R,G,B;
R = lRGBColor.rgbRed;
G = lRGBColor.rgbGreen;
B = lRGBColor.rgbBlue;
// Y = (int)( 0.257 * R + 0.504 * G + 0.098 * B);
// U = (int)( 0.439 * R - 0.368 * G - 0.071 * B + 128);
// V = (int)(-0.148 * R - 0.291 * G + 0.439 * B + 128);
Y = (int)(0.299f * R + 0.587f * G + 0.114f * B);
U = (int)((B-Y) * 0.565f + 128);
V = (int)((R-Y) * 0.713f + 128);
Y= min(255,max(0,Y));
U= min(255,max(0,U));
V= min(255,max(0,V));
RGBQUAD yuv={(BYTE)V,(BYTE)U,(BYTE)Y,0};
return yuv;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::YIQtoRGB(RGBQUAD lYIQColor)
{
int I,Q,R,G,B;
float Y = lYIQColor.rgbRed;
I = lYIQColor.rgbGreen - 128;
Q = lYIQColor.rgbBlue - 128;
R = (int)( Y + 0.956f * I + 0.621f * Q);
G = (int)( Y - 0.273f * I - 0.647f * Q);
B = (int)( Y - 1.104f * I + 1.701f * Q);
R= min(255,max(0,R));
G= min(255,max(0,G));
B= min(255,max(0,B));
RGBQUAD rgb={(BYTE)B,(BYTE)G,(BYTE)R,0};
return rgb;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::RGBtoYIQ(RGBQUAD lRGBColor)
{
int Y,I,Q,R,G,B;
R = lRGBColor.rgbRed;
G = lRGBColor.rgbGreen;
B = lRGBColor.rgbBlue;
Y = (int)( 0.2992f * R + 0.5868f * G + 0.1140f * B);
I = (int)( 0.5960f * R - 0.2742f * G - 0.3219f * B + 128);
Q = (int)( 0.2109f * R - 0.5229f * G + 0.3120f * B + 128);
Y= min(255,max(0,Y));
I= min(255,max(0,I));
Q= min(255,max(0,Q));
RGBQUAD yiq={(BYTE)Q,(BYTE)I,(BYTE)Y,0};
return yiq;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::XYZtoRGB(RGBQUAD lXYZColor)
{
int X,Y,Z,R,G,B;
X = lXYZColor.rgbRed;
Y = lXYZColor.rgbGreen;
Z = lXYZColor.rgbBlue;
double k=1.088751;
R = (int)( 3.240479f * X - 1.537150f * Y - 0.498535f * Z * k);
G = (int)( -0.969256f * X + 1.875992f * Y + 0.041556f * Z * k);
B = (int)( 0.055648f * X - 0.204043f * Y + 1.057311f * Z * k);
R= min(255,max(0,R));
G= min(255,max(0,G));
B= min(255,max(0,B));
RGBQUAD rgb={(BYTE)B,(BYTE)G,(BYTE)R,0};
return rgb;
}
////////////////////////////////////////////////////////////////////////////////
RGBQUAD CxImage::RGBtoXYZ(RGBQUAD lRGBColor)
{
int X,Y,Z,R,G,B;
R = lRGBColor.rgbRed;
G = lRGBColor.rgbGreen;
B = lRGBColor.rgbBlue;
X = (int)( 0.412453f * R + 0.357580f * G + 0.180423f * B);
Y = (int)( 0.212671f * R + 0.715160f * G + 0.072169f * B);
Z = (int)((0.019334f * R + 0.119193f * G + 0.950227f * B)*0.918483657f);
//X= min(255,max(0,X));
//Y= min(255,max(0,Y));
//Z= min(255,max(0,Z));
RGBQUAD xyz={(BYTE)Z,(BYTE)Y,(BYTE)X,0};
return xyz;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Generates a "rainbow" palette with saturated colors
* \param correction: 1 generates a single hue spectrum. 0.75 is nice for scientific applications.
*/
void CxImage::HuePalette(float correction)
{
if (head.biClrUsed==0) return;
for(DWORD j=0; j<head.biClrUsed; j++){
BYTE i=(BYTE)(j*correction*(255/(head.biClrUsed-1)));
RGBQUAD hsl={120,240,i,0};
SetPaletteColor((BYTE)j,HSLtoRGB(hsl));
}
}
////////////////////////////////////////////////////////////////////////////////
/**
* Replaces the original hue and saturation values.
* \param hue: hue
* \param sat: saturation
* \param blend: can be from 0 (no effect) to 1 (full effect)
* \return true if everything is ok
*/
bool CxImage::Colorize(BYTE hue, BYTE sat, float blend)
{
if (!pDib) return false;
if (blend < 0.0f) blend = 0.0f;
if (blend > 1.0f) blend = 1.0f;
int a0 = (int)(256*blend);
int a1 = 256 - a0;
bool bFullBlend = false;
if (blend > 0.999f) bFullBlend = true;
RGBQUAD color,hsl;
if (head.biClrUsed==0){
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
if (bFullBlend){
color = RGBtoHSL(BlindGetPixelColor(x,y));
color.rgbRed=hue;
color.rgbGreen=sat;
BlindSetPixelColor(x,y,HSLtoRGB(color));
} else {
color = BlindGetPixelColor(x,y);
hsl.rgbRed=hue;
hsl.rgbGreen=sat;
hsl.rgbBlue = (BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue);
hsl = HSLtoRGB(hsl);
//BlendPixelColor(x,y,hsl,blend);
//color.rgbRed = (BYTE)(hsl.rgbRed * blend + color.rgbRed * (1.0f - blend));
//color.rgbBlue = (BYTE)(hsl.rgbBlue * blend + color.rgbBlue * (1.0f - blend));
//color.rgbGreen = (BYTE)(hsl.rgbGreen * blend + color.rgbGreen * (1.0f - blend));
color.rgbRed = (BYTE)((hsl.rgbRed * a0 + color.rgbRed * a1)>>8);
color.rgbBlue = (BYTE)((hsl.rgbBlue * a0 + color.rgbBlue * a1)>>8);
color.rgbGreen = (BYTE)((hsl.rgbGreen * a0 + color.rgbGreen * a1)>>8);
BlindSetPixelColor(x,y,color);
}
}
}
}
} else {
for(DWORD j=0; j<head.biClrUsed; j++){
if (bFullBlend){
color = RGBtoHSL(GetPaletteColor((BYTE)j));
color.rgbRed=hue;
color.rgbGreen=sat;
SetPaletteColor((BYTE)j,HSLtoRGB(color));
} else {
color = GetPaletteColor((BYTE)j);
hsl.rgbRed=hue;
hsl.rgbGreen=sat;
hsl.rgbBlue = (BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue);
hsl = HSLtoRGB(hsl);
color.rgbRed = (BYTE)(hsl.rgbRed * blend + color.rgbRed * (1.0f - blend));
color.rgbBlue = (BYTE)(hsl.rgbBlue * blend + color.rgbBlue * (1.0f - blend));
color.rgbGreen = (BYTE)(hsl.rgbGreen * blend + color.rgbGreen * (1.0f - blend));
SetPaletteColor((BYTE)j,color);
}
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Changes the brightness and the contrast of the image.
* \param brightness: can be from -255 to 255, if brightness is negative, the image becomes dark.
* \param contrast: can be from -100 to 100, the neutral value is 0.
* \return true if everything is ok
*/
bool CxImage::Light(long brightness, long contrast)
{
if (!pDib) return false;
float c=(100 + contrast)/100.0f;
brightness+=128;
BYTE cTable[256]; //<nipper>
for (int i=0;i<256;i++) {
cTable[i] = (BYTE)max(0,min(255,(int)((i-128)*c + brightness + 0.5f)));
}
return Lut(cTable);
}
////////////////////////////////////////////////////////////////////////////////
/**
* \return mean lightness of the image. Useful with Threshold() and Light()
*/
float CxImage::Mean()
{
if (!pDib) return 0;
CxImage tmp(*this,true);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
tmp.GrayScale();
float sum=0;
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin==xmax || ymin==ymax) return (float)0.0;
BYTE *iSrc=tmp.info.pImage;
iSrc += tmp.info.dwEffWidth*ymin; // necessary for selections <Admir Hodzic>
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin)); //<zhanghk><Anatoly Ivasyuk>
for(long x=xmin; x<xmax; x++){
sum+=iSrc[x];
}
iSrc+=tmp.info.dwEffWidth;
}
return sum/(xmax-xmin)/(ymax-ymin);
}
////////////////////////////////////////////////////////////////////////////////
/**
* 2D linear filter
* \param kernel: convolving matrix, in row format.
* \param Ksize: size of the kernel.
* \param Kfactor: normalization constant.
* \param Koffset: bias.
* \verbatim Example: the "soften" filter uses this kernel:
1 1 1
1 8 1
1 1 1
the function needs: kernel={1,1,1,1,8,1,1,1,1}; Ksize=3; Kfactor=16; Koffset=0; \endverbatim
* \return true if everything is ok
*/
bool CxImage::Filter(long* kernel, long Ksize, long Kfactor, long Koffset)
{
if (!pDib) return false;
long k2 = Ksize/2;
long kmax= Ksize-k2;
long r,g,b,i;
long ksumcur,ksumtot;
RGBQUAD c;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
ksumtot = 0;
for(long j=-k2;j<kmax;j++){
for(long k=-k2;k<kmax;k++){
ksumtot += kernel[(j+k2)+Ksize*(k+k2)];
}
}
if ((head.biBitCount==8) && IsGrayScale())
{
unsigned char* cPtr;
unsigned char* cPtr2;
int iCount;
int iY, iY2, iY1;
cPtr = info.pImage;
cPtr2 = (unsigned char *)tmp.info.pImage;
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
iY1 = y*info.dwEffWidth+xmin;
for(long x=xmin; x<xmax; x++, iY1++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
b=ksumcur=0;
iCount = 0;
iY2 = ((y-k2)*info.dwEffWidth);
for(long j=-k2;j<kmax;j++, iY2+=info.dwEffWidth)
{
if (0>(y+j) || (y+j)>=head.biHeight) continue;
iY = iY2+x;
for(long k=-k2;k<kmax;k++, iCount++)
{
if (0>(x+k) || (x+k)>=head.biWidth) continue;
i=kernel[iCount];
b += cPtr[iY+k] * i;
ksumcur += i;
}
}
if (Kfactor==0 || ksumcur==0){
cPtr2[iY1] = (BYTE)min(255, max(0,(int)(b + Koffset)));
} else if (ksumtot == ksumcur) {
cPtr2[iY1] = (BYTE)min(255, max(0,(int)(b/Kfactor + Koffset)));
} else {
cPtr2[iY1] = (BYTE)min(255, max(0,(int)((b*ksumtot)/(ksumcur*Kfactor) + Koffset)));
}
}
}
}
}
else
{
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
r=b=g=ksumcur=0;
for(long j=-k2;j<kmax;j++){
for(long k=-k2;k<kmax;k++){
if (!IsInside(x+j,y+k)) continue;
c = BlindGetPixelColor(x+j,y+k);
i = kernel[(j+k2)+Ksize*(k+k2)];
r += c.rgbRed * i;
g += c.rgbGreen * i;
b += c.rgbBlue * i;
ksumcur += i;
}
}
if (Kfactor==0 || ksumcur==0){
c.rgbRed = (BYTE)min(255, max(0,(int)(r + Koffset)));
c.rgbGreen = (BYTE)min(255, max(0,(int)(g + Koffset)));
c.rgbBlue = (BYTE)min(255, max(0,(int)(b + Koffset)));
} else if (ksumtot == ksumcur) {
c.rgbRed = (BYTE)min(255, max(0,(int)(r/Kfactor + Koffset)));
c.rgbGreen = (BYTE)min(255, max(0,(int)(g/Kfactor + Koffset)));
c.rgbBlue = (BYTE)min(255, max(0,(int)(b/Kfactor + Koffset)));
} else {
c.rgbRed = (BYTE)min(255, max(0,(int)((r*ksumtot)/(ksumcur*Kfactor) + Koffset)));
c.rgbGreen = (BYTE)min(255, max(0,(int)((g*ksumtot)/(ksumcur*Kfactor) + Koffset)));
c.rgbBlue = (BYTE)min(255, max(0,(int)((b*ksumtot)/(ksumcur*Kfactor) + Koffset)));
}
tmp.BlindSetPixelColor(x,y,c);
}
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Enhance the dark areas of the image
* \param Ksize: size of the kernel.
* \return true if everything is ok
*/
bool CxImage::Erode(long Ksize)
{
if (!pDib) return false;
long k2 = Ksize/2;
long kmax= Ksize-k2;
BYTE r,g,b;
RGBQUAD c;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
r=b=g=255;
for(long j=-k2;j<kmax;j++){
for(long k=-k2;k<kmax;k++){
if (!IsInside(x+j,y+k)) continue;
c = BlindGetPixelColor(x+j,y+k);
if (c.rgbRed < r) r=c.rgbRed;
if (c.rgbGreen < g) g=c.rgbGreen;
if (c.rgbBlue < b) b=c.rgbBlue;
}
}
c.rgbRed = r;
c.rgbGreen = g;
c.rgbBlue = b;
tmp.BlindSetPixelColor(x,y,c);
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Enhance the light areas of the image
* \param Ksize: size of the kernel.
* \return true if everything is ok
*/
bool CxImage::Dilate(long Ksize)
{
if (!pDib) return false;
long k2 = Ksize/2;
long kmax= Ksize-k2;
BYTE r,g,b;
RGBQUAD c;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
r=b=g=0;
for(long j=-k2;j<kmax;j++){
for(long k=-k2;k<kmax;k++){
if (!IsInside(x+j,y+k)) continue;
c = BlindGetPixelColor(x+j,y+k);
if (c.rgbRed > r) r=c.rgbRed;
if (c.rgbGreen > g) g=c.rgbGreen;
if (c.rgbBlue > b) b=c.rgbBlue;
}
}
c.rgbRed = r;
c.rgbGreen = g;
c.rgbBlue = b;
tmp.BlindSetPixelColor(x,y,c);
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Enhance the variations between adjacent pixels.
* Similar results can be achieved using Filter(),
* but the algorithms are different both in Edge() and in Contour().
* \param Ksize: size of the kernel.
* \return true if everything is ok
*/
bool CxImage::Edge(long Ksize)
{
if (!pDib) return false;
long k2 = Ksize/2;
long kmax= Ksize-k2;
BYTE r,g,b,rr,gg,bb;
RGBQUAD c;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
r=b=g=0;
rr=bb=gg=255;
for(long j=-k2;j<kmax;j++){
for(long k=-k2;k<kmax;k++){
if (!IsInside(x+j,y+k)) continue;
c = BlindGetPixelColor(x+j,y+k);
if (c.rgbRed > r) r=c.rgbRed;
if (c.rgbGreen > g) g=c.rgbGreen;
if (c.rgbBlue > b) b=c.rgbBlue;
if (c.rgbRed < rr) rr=c.rgbRed;
if (c.rgbGreen < gg) gg=c.rgbGreen;
if (c.rgbBlue < bb) bb=c.rgbBlue;
}
}
c.rgbRed = (BYTE)(255-abs(r-rr));
c.rgbGreen = (BYTE)(255-abs(g-gg));
c.rgbBlue = (BYTE)(255-abs(b-bb));
tmp.BlindSetPixelColor(x,y,c);
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Blends two images
* \param imgsrc2: image to be mixed with this
* \param op: blending method; see ImageOpType
* \param lXOffset, lYOffset: image displacement
* \param bMixAlpha: if true and imgsrc2 has a valid alpha layer, it will be mixed in the destination image.
* \return true if everything is ok
*
* thanks to Mwolski
*/
//
void CxImage::Mix(CxImage & imgsrc2, ImageOpType op, long lXOffset, long lYOffset, bool bMixAlpha)
{
long lWide = min(GetWidth(),imgsrc2.GetWidth()-lXOffset);
long lHeight = min(GetHeight(),imgsrc2.GetHeight()-lYOffset);
bool bEditAlpha = imgsrc2.AlphaIsValid() & bMixAlpha;
if (bEditAlpha && AlphaIsValid()==false){
AlphaCreate();
}
RGBQUAD rgbBackgrnd1 = GetTransColor();
RGBQUAD rgb1, rgb2, rgbDest;
for(long lY=0;lY<lHeight;lY++)
{
info.nProgress = (long)(100*lY/head.biHeight);
if (info.nEscape) break;
for(long lX=0;lX<lWide;lX++)
{
#if CXIMAGE_SUPPORT_SELECTION
if (SelectionIsInside(lX,lY) && imgsrc2.SelectionIsInside(lX+lXOffset,lY+lYOffset))
#endif //CXIMAGE_SUPPORT_SELECTION
{
rgb1 = GetPixelColor(lX,lY);
rgb2 = imgsrc2.GetPixelColor(lX+lXOffset,lY+lYOffset);
switch(op)
{
case OpAvg:
rgbDest.rgbBlue = (BYTE)((rgb1.rgbBlue+rgb2.rgbBlue)/2);
rgbDest.rgbGreen = (BYTE)((rgb1.rgbGreen+rgb2.rgbGreen)/2);
rgbDest.rgbRed = (BYTE)((rgb1.rgbRed+rgb2.rgbRed)/2);
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)((rgb1.rgbReserved+rgb2.rgbReserved)/2);
break;
case OpAdd:
rgbDest.rgbBlue = (BYTE)max(0,min(255,rgb1.rgbBlue+rgb2.rgbBlue));
rgbDest.rgbGreen = (BYTE)max(0,min(255,rgb1.rgbGreen+rgb2.rgbGreen));
rgbDest.rgbRed = (BYTE)max(0,min(255,rgb1.rgbRed+rgb2.rgbRed));
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)max(0,min(255,rgb1.rgbReserved+rgb2.rgbReserved));
break;
case OpSub:
rgbDest.rgbBlue = (BYTE)max(0,min(255,rgb1.rgbBlue-rgb2.rgbBlue));
rgbDest.rgbGreen = (BYTE)max(0,min(255,rgb1.rgbGreen-rgb2.rgbGreen));
rgbDest.rgbRed = (BYTE)max(0,min(255,rgb1.rgbRed-rgb2.rgbRed));
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)max(0,min(255,rgb1.rgbReserved-rgb2.rgbReserved));
break;
case OpAnd:
rgbDest.rgbBlue = (BYTE)(rgb1.rgbBlue&rgb2.rgbBlue);
rgbDest.rgbGreen = (BYTE)(rgb1.rgbGreen&rgb2.rgbGreen);
rgbDest.rgbRed = (BYTE)(rgb1.rgbRed&rgb2.rgbRed);
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)(rgb1.rgbReserved&rgb2.rgbReserved);
break;
case OpXor:
rgbDest.rgbBlue = (BYTE)(rgb1.rgbBlue^rgb2.rgbBlue);
rgbDest.rgbGreen = (BYTE)(rgb1.rgbGreen^rgb2.rgbGreen);
rgbDest.rgbRed = (BYTE)(rgb1.rgbRed^rgb2.rgbRed);
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)(rgb1.rgbReserved^rgb2.rgbReserved);
break;
case OpOr:
rgbDest.rgbBlue = (BYTE)(rgb1.rgbBlue|rgb2.rgbBlue);
rgbDest.rgbGreen = (BYTE)(rgb1.rgbGreen|rgb2.rgbGreen);
rgbDest.rgbRed = (BYTE)(rgb1.rgbRed|rgb2.rgbRed);
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)(rgb1.rgbReserved|rgb2.rgbReserved);
break;
case OpMask:
if(rgb2.rgbBlue==0 && rgb2.rgbGreen==0 && rgb2.rgbRed==0)
rgbDest = rgbBackgrnd1;
else
rgbDest = rgb1;
break;
case OpSrcCopy:
if(IsTransparent(lX,lY))
rgbDest = rgb2;
else // copy straight over
rgbDest = rgb1;
break;
case OpDstCopy:
if(imgsrc2.IsTransparent(lX+lXOffset,lY+lYOffset))
rgbDest = rgb1;
else // copy straight over
rgbDest = rgb2;
break;
case OpScreen:
{
BYTE a,a1;
if (imgsrc2.IsTransparent(lX+lXOffset,lY+lYOffset)){
a=0;
} else if (imgsrc2.AlphaIsValid()){
a=imgsrc2.AlphaGet(lX+lXOffset,lY+lYOffset);
a =(BYTE)((a*imgsrc2.info.nAlphaMax)/255);
} else {
a=255;
}
if (a==0){ //transparent
rgbDest = rgb1;
} else if (a==255){ //opaque
rgbDest = rgb2;
} else { //blend
a1 = (BYTE)~a;
rgbDest.rgbBlue = (BYTE)((rgb1.rgbBlue*a1+rgb2.rgbBlue*a)/255);
rgbDest.rgbGreen = (BYTE)((rgb1.rgbGreen*a1+rgb2.rgbGreen*a)/255);
rgbDest.rgbRed = (BYTE)((rgb1.rgbRed*a1+rgb2.rgbRed*a)/255);
}
if (bEditAlpha) rgbDest.rgbReserved = (BYTE)((rgb1.rgbReserved*a)/255);
}
break;
case OpSrcBlend:
if(IsTransparent(lX,lY))
rgbDest = rgb2;
else
{
long lBDiff = abs(rgb1.rgbBlue - rgbBackgrnd1.rgbBlue);
long lGDiff = abs(rgb1.rgbGreen - rgbBackgrnd1.rgbGreen);
long lRDiff = abs(rgb1.rgbRed - rgbBackgrnd1.rgbRed);
double lAverage = (lBDiff+lGDiff+lRDiff)/3;
double lThresh = 16;
double dLarge = lAverage/lThresh;
double dSmall = (lThresh-lAverage)/lThresh;
double dSmallAmt = dSmall*((double)rgb2.rgbBlue);
if( lAverage < lThresh+1){
rgbDest.rgbBlue = (BYTE)max(0,min(255,(int)(dLarge*((double)rgb1.rgbBlue) +
dSmallAmt)));
rgbDest.rgbGreen = (BYTE)max(0,min(255,(int)(dLarge*((double)rgb1.rgbGreen) +
dSmallAmt)));
rgbDest.rgbRed = (BYTE)max(0,min(255,(int)(dLarge*((double)rgb1.rgbRed) +
dSmallAmt)));
}
else
rgbDest = rgb1;
}
break;
default:
return;
}
SetPixelColor(lX,lY,rgbDest,bEditAlpha);
}
}
}
}
////////////////////////////////////////////////////////////////////////////////
// thanks to Kenneth Ballard
void CxImage::MixFrom(CxImage & imagesrc2, long lXOffset, long lYOffset)
{
long width = imagesrc2.GetWidth();
long height = imagesrc2.GetHeight();
int x, y;
if (imagesrc2.IsTransparent()) {
for(x = 0; x < width; x++) {
for(y = 0; y < height; y++) {
if(!imagesrc2.IsTransparent(x,y)){
SetPixelColor(x + lXOffset, y + lYOffset, imagesrc2.BlindGetPixelColor(x, y));
}
}
}
} else { //no transparency so just set it <Matt>
for(x = 0; x < width; x++) {
for(y = 0; y < height; y++) {
SetPixelColor(x + lXOffset, y + lYOffset, imagesrc2.BlindGetPixelColor(x, y));
}
}
}
}
////////////////////////////////////////////////////////////////////////////////
/**
* Adjusts separately the red, green, and blue values in the image.
* \param r, g, b: can be from -255 to +255.
* \return true if everything is ok
*/
bool CxImage::ShiftRGB(long r, long g, long b)
{
if (!pDib) return false;
RGBQUAD color;
if (head.biClrUsed==0){
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
color = BlindGetPixelColor(x,y);
color.rgbRed = (BYTE)max(0,min(255,(int)(color.rgbRed + r)));
color.rgbGreen = (BYTE)max(0,min(255,(int)(color.rgbGreen + g)));
color.rgbBlue = (BYTE)max(0,min(255,(int)(color.rgbBlue + b)));
BlindSetPixelColor(x,y,color);
}
}
}
} else {
for(DWORD j=0; j<head.biClrUsed; j++){
color = GetPaletteColor((BYTE)j);
color.rgbRed = (BYTE)max(0,min(255,(int)(color.rgbRed + r)));
color.rgbGreen = (BYTE)max(0,min(255,(int)(color.rgbGreen + g)));
color.rgbBlue = (BYTE)max(0,min(255,(int)(color.rgbBlue + b)));
SetPaletteColor((BYTE)j,color);
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Adjusts the color balance of the image
* \param gamma can be from 0.1 to 5.
* \return true if everything is ok
* \sa GammaRGB
*/
bool CxImage::Gamma(float gamma)
{
if (!pDib) return false;
if (gamma <= 0.0f) return false;
double dinvgamma = 1/gamma;
double dMax = pow(255.0, dinvgamma) / 255.0;
BYTE cTable[256]; //<nipper>
for (int i=0;i<256;i++) {
cTable[i] = (BYTE)max(0,min(255,(int)( pow((double)i, dinvgamma) / dMax)));
}
return Lut(cTable);
}
////////////////////////////////////////////////////////////////////////////////
/**
* Adjusts the color balance indipendent for each color channel
* \param gammaR, gammaG, gammaB can be from 0.1 to 5.
* \return true if everything is ok
* \sa Gamma
*/
bool CxImage::GammaRGB(float gammaR, float gammaG, float gammaB)
{
if (!pDib) return false;
if (gammaR <= 0.0f) return false;
if (gammaG <= 0.0f) return false;
if (gammaB <= 0.0f) return false;
double dinvgamma, dMax;
int i;
dinvgamma = 1/gammaR;
dMax = pow(255.0, dinvgamma) / 255.0;
BYTE cTableR[256];
for (i=0;i<256;i++) {
cTableR[i] = (BYTE)max(0,min(255,(int)( pow((double)i, dinvgamma) / dMax)));
}
dinvgamma = 1/gammaG;
dMax = pow(255.0, dinvgamma) / 255.0;
BYTE cTableG[256];
for (i=0;i<256;i++) {
cTableG[i] = (BYTE)max(0,min(255,(int)( pow((double)i, dinvgamma) / dMax)));
}
dinvgamma = 1/gammaB;
dMax = pow(255.0, dinvgamma) / 255.0;
BYTE cTableB[256];
for (i=0;i<256;i++) {
cTableB[i] = (BYTE)max(0,min(255,(int)( pow((double)i, dinvgamma) / dMax)));
}
return Lut(cTableR, cTableG, cTableB);
}
////////////////////////////////////////////////////////////////////////////////
//#if !defined (_WIN32_WCE)
/**
* Adjusts the intensity of each pixel to the median intensity of its surrounding pixels.
* \param Ksize: size of the kernel.
* \return true if everything is ok
*/
bool CxImage::Median(long Ksize)
{
if (!pDib) return false;
long k2 = Ksize/2;
long kmax= Ksize-k2;
long i,j,k;
RGBQUAD* kernel = (RGBQUAD*)malloc(Ksize*Ksize*sizeof(RGBQUAD));
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
for(j=-k2, i=0;j<kmax;j++)
for(k=-k2;k<kmax;k++)
if (IsInside(x+j,y+k))
kernel[i++]=BlindGetPixelColor(x+j,y+k);
qsort(kernel, i, sizeof(RGBQUAD), CompareColors);
tmp.SetPixelColor(x,y,kernel[i/2]);
}
}
}
free(kernel);
Transfer(tmp);
return true;
}
//#endif //_WIN32_WCE
////////////////////////////////////////////////////////////////////////////////
/**
* Adds an uniform noise to the image
* \param level: can be from 0 (no noise) to 255 (lot of noise).
* \return true if everything is ok
*/
bool CxImage::Noise(long level)
{
if (!pDib) return false;
RGBQUAD color;
long xmin,xmax,ymin,ymax,n;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin)); //<zhanghk><Anatoly Ivasyuk>
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
color = BlindGetPixelColor(x,y);
n=(long)((rand()/(float)RAND_MAX - 0.5)*level);
color.rgbRed = (BYTE)max(0,min(255,(int)(color.rgbRed + n)));
n=(long)((rand()/(float)RAND_MAX - 0.5)*level);
color.rgbGreen = (BYTE)max(0,min(255,(int)(color.rgbGreen + n)));
n=(long)((rand()/(float)RAND_MAX - 0.5)*level);
color.rgbBlue = (BYTE)max(0,min(255,(int)(color.rgbBlue + n)));
BlindSetPixelColor(x,y,color);
}
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Computes the bidimensional FFT or DFT of the image.
* - The images are processed as grayscale
* - If the dimensions of the image are a power of, 2 the FFT is performed automatically.
* - If dstReal and/or dstImag are NULL, the resulting images replaces the original(s).
* - Note: with 8 bits there is a HUGE loss in the dynamics. The function tries
* to keep an acceptable SNR, but 8bit = 48dB...
*
* \param srcReal, srcImag: source images: One can be NULL, but not both
* \param dstReal, dstImag: destination images. Can be NULL.
* \param direction: 1 = forward, -1 = inverse.
* \param bForceFFT: if true, the images are resampled to make the dimensions a power of 2.
* \param bMagnitude: if true, the real part returns the magnitude, the imaginary part returns the phase
* \return true if everything is ok
*/
bool CxImage::FFT2(CxImage* srcReal, CxImage* srcImag, CxImage* dstReal, CxImage* dstImag,
long direction, bool bForceFFT, bool bMagnitude)
{
//check if there is something to convert
if (srcReal==NULL && srcImag==NULL) return false;
long w,h;
//get width and height
if (srcReal) {
w=srcReal->GetWidth();
h=srcReal->GetHeight();
} else {
w=srcImag->GetWidth();
h=srcImag->GetHeight();
}
bool bXpow2 = IsPowerof2(w);
bool bYpow2 = IsPowerof2(h);
//if bForceFFT, width AND height must be powers of 2
if (bForceFFT && !(bXpow2 && bYpow2)) {
long i;
i=0;
while((1<<i)<w) i++;
w=1<<i;
bXpow2=true;
i=0;
while((1<<i)<h) i++;
h=1<<i;
bYpow2=true;
}
// I/O images for FFT
CxImage *tmpReal,*tmpImag;
// select output
tmpReal = (dstReal) ? dstReal : srcReal;
tmpImag = (dstImag) ? dstImag : srcImag;
// src!=dst -> copy the image
if (srcReal && dstReal) tmpReal->Copy(*srcReal,true,false,false);
if (srcImag && dstImag) tmpImag->Copy(*srcImag,true,false,false);
// dst&&src are empty -> create new one, else turn to GrayScale
if (srcReal==0 && dstReal==0){
tmpReal = new CxImage(w,h,8);
tmpReal->Clear(0);
tmpReal->SetGrayPalette();
} else {
if (!tmpReal->IsGrayScale()) tmpReal->GrayScale();
}
if (srcImag==0 && dstImag==0){
tmpImag = new CxImage(w,h,8);
tmpImag->Clear(0);
tmpImag->SetGrayPalette();
} else {
if (!tmpImag->IsGrayScale()) tmpImag->GrayScale();
}
if (!(tmpReal->IsValid() && tmpImag->IsValid())){
if (srcReal==0 && dstReal==0) delete tmpReal;
if (srcImag==0 && dstImag==0) delete tmpImag;
return false;
}
//resample for FFT, if necessary
tmpReal->Resample(w,h,0);
tmpImag->Resample(w,h,0);
//ok, here we have 2 (w x h), grayscale images ready for a FFT
double* real;
double* imag;
long j,k,m;
_complex **grid;
//double mean = tmpReal->Mean();
/* Allocate memory for the grid */
grid = (_complex **)malloc(w * sizeof(_complex));
for (k=0;k<w;k++) {
grid[k] = (_complex *)malloc(h * sizeof(_complex));
}
for (j=0;j<h;j++) {
for (k=0;k<w;k++) {
grid[k][j].x = tmpReal->GetPixelIndex(k,j)-128;
grid[k][j].y = tmpImag->GetPixelIndex(k,j)-128;
}
}
//DFT buffers
double *real2,*imag2;
real2 = (double*)malloc(max(w,h) * sizeof(double));
imag2 = (double*)malloc(max(w,h) * sizeof(double));
/* Transform the rows */
real = (double *)malloc(w * sizeof(double));
imag = (double *)malloc(w * sizeof(double));
m=0;
while((1<<m)<w) m++;
for (j=0;j<h;j++) {
for (k=0;k<w;k++) {
real[k] = grid[k][j].x;
imag[k] = grid[k][j].y;
}
if (bXpow2) FFT(direction,m,real,imag);
else DFT(direction,w,real,imag,real2,imag2);
for (k=0;k<w;k++) {
grid[k][j].x = real[k];
grid[k][j].y = imag[k];
}
}
free(real);
free(imag);
/* Transform the columns */
real = (double *)malloc(h * sizeof(double));
imag = (double *)malloc(h * sizeof(double));
m=0;
while((1<<m)<h) m++;
for (k=0;k<w;k++) {
for (j=0;j<h;j++) {
real[j] = grid[k][j].x;
imag[j] = grid[k][j].y;
}
if (bYpow2) FFT(direction,m,real,imag);
else DFT(direction,h,real,imag,real2,imag2);
for (j=0;j<h;j++) {
grid[k][j].x = real[j];
grid[k][j].y = imag[j];
}
}
free(real);
free(imag);
free(real2);
free(imag2);
/* converting from double to byte, there is a HUGE loss in the dynamics
"nn" tries to keep an acceptable SNR, but 8bit=48dB: don't ask more */
double nn=pow((double)2,(double)log((double)max(w,h))/(double)log((double)2)-4);
//reversed gain for reversed transform
if (direction==-1) nn=1/nn;
//bMagnitude : just to see it on the screen
if (bMagnitude) nn*=4;
for (j=0;j<h;j++) {
for (k=0;k<w;k++) {
if (bMagnitude){
tmpReal->SetPixelIndex(k,j,(BYTE)max(0,min(255,(nn*(3+log(_cabs(grid[k][j])))))));
if (grid[k][j].x==0){
tmpImag->SetPixelIndex(k,j,(BYTE)max(0,min(255,(128+(atan(grid[k][j].y/0.0000000001)*nn)))));
} else {
tmpImag->SetPixelIndex(k,j,(BYTE)max(0,min(255,(128+(atan(grid[k][j].y/grid[k][j].x)*nn)))));
}
} else {
tmpReal->SetPixelIndex(k,j,(BYTE)max(0,min(255,(128 + grid[k][j].x*nn))));
tmpImag->SetPixelIndex(k,j,(BYTE)max(0,min(255,(128 + grid[k][j].y*nn))));
}
}
}
for (k=0;k<w;k++) free (grid[k]);
free (grid);
if (srcReal==0 && dstReal==0) delete tmpReal;
if (srcImag==0 && dstImag==0) delete tmpImag;
return true;
}
////////////////////////////////////////////////////////////////////////////////
bool CxImage::IsPowerof2(long x)
{
long i=0;
while ((1<<i)<x) i++;
if (x==(1<<i)) return true;
return false;
}
////////////////////////////////////////////////////////////////////////////////
/**
This computes an in-place complex-to-complex FFT
x and y are the real and imaginary arrays of n=2^m points.
o(n)=n*log2(n)
dir = 1 gives forward transform
dir = -1 gives reverse transform
Written by Paul Bourke, July 1998
FFT algorithm by Cooley and Tukey, 1965
*/
bool CxImage::FFT(int dir,int m,double *x,double *y)
{
long nn,i,i1,j,k,i2,l,l1,l2;
double c1,c2,tx,ty,t1,t2,u1,u2,z;
/* Calculate the number of points */
nn = 1<<m;
/* Do the bit reversal */
i2 = nn >> 1;
j = 0;
for (i=0;i<nn-1;i++) {
if (i < j) {
tx = x[i];
ty = y[i];
x[i] = x[j];
y[i] = y[j];
x[j] = tx;
y[j] = ty;
}
k = i2;
while (k <= j) {
j -= k;
k >>= 1;
}
j += k;
}
/* Compute the FFT */
c1 = -1.0;
c2 = 0.0;
l2 = 1;
for (l=0;l<m;l++) {
l1 = l2;
l2 <<= 1;
u1 = 1.0;
u2 = 0.0;
for (j=0;j<l1;j++) {
for (i=j;i<nn;i+=l2) {
i1 = i + l1;
t1 = u1 * x[i1] - u2 * y[i1];
t2 = u1 * y[i1] + u2 * x[i1];
x[i1] = x[i] - t1;
y[i1] = y[i] - t2;
x[i] += t1;
y[i] += t2;
}
z = u1 * c1 - u2 * c2;
u2 = u1 * c2 + u2 * c1;
u1 = z;
}
c2 = sqrt((1.0 - c1) / 2.0);
if (dir == 1)
c2 = -c2;
c1 = sqrt((1.0 + c1) / 2.0);
}
/* Scaling for forward transform */
if (dir == 1) {
for (i=0;i<nn;i++) {
x[i] /= (double)nn;
y[i] /= (double)nn;
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
Direct fourier transform o(n)=n^2
Written by Paul Bourke, July 1998
*/
bool CxImage::DFT(int dir,long m,double *x1,double *y1,double *x2,double *y2)
{
long i,k;
double arg;
double cosarg,sinarg;
for (i=0;i<m;i++) {
x2[i] = 0;
y2[i] = 0;
arg = - dir * 2.0 * PI * i / (double)m;
for (k=0;k<m;k++) {
cosarg = cos(k * arg);
sinarg = sin(k * arg);
x2[i] += (x1[k] * cosarg - y1[k] * sinarg);
y2[i] += (x1[k] * sinarg + y1[k] * cosarg);
}
}
/* Copy the data back */
if (dir == 1) {
for (i=0;i<m;i++) {
x1[i] = x2[i] / m;
y1[i] = y2[i] / m;
}
} else {
for (i=0;i<m;i++) {
x1[i] = x2[i];
y1[i] = y2[i];
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Combines different color components into a single image
* \param r,g,b: color channels
* \param a: alpha layer, can be NULL
* \param colorspace: 0 = RGB, 1 = HSL, 2 = YUV, 3 = YIQ, 4 = XYZ
* \return true if everything is ok
*/
bool CxImage::Combine(CxImage* r,CxImage* g,CxImage* b,CxImage* a, long colorspace)
{
if (r==0 || g==0 || b==0) return false;
long w = r->GetWidth();
long h = r->GetHeight();
Create(w,h,24);
g->Resample(w,h);
b->Resample(w,h);
if (a) {
a->Resample(w,h);
#if CXIMAGE_SUPPORT_ALPHA
AlphaCreate();
#endif //CXIMAGE_SUPPORT_ALPHA
}
RGBQUAD c;
for (long y=0;y<h;y++){
info.nProgress = (long)(100*y/h); //<Anatoly Ivasyuk>
for (long x=0;x<w;x++){
c.rgbRed=r->GetPixelIndex(x,y);
c.rgbGreen=g->GetPixelIndex(x,y);
c.rgbBlue=b->GetPixelIndex(x,y);
switch (colorspace){
case 1:
BlindSetPixelColor(x,y,HSLtoRGB(c));
break;
case 2:
BlindSetPixelColor(x,y,YUVtoRGB(c));
break;
case 3:
BlindSetPixelColor(x,y,YIQtoRGB(c));
break;
case 4:
BlindSetPixelColor(x,y,XYZtoRGB(c));
break;
default:
BlindSetPixelColor(x,y,c);
}
#if CXIMAGE_SUPPORT_ALPHA
if (a) AlphaSet(x,y,a->GetPixelIndex(x,y));
#endif //CXIMAGE_SUPPORT_ALPHA
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Smart blurring to remove small defects, dithering or artifacts.
* \param radius: normally between 0.01 and 0.5
* \param niterations: should be trimmed with radius, to avoid blurring should be (radius*niterations)<1
* \param colorspace: 0 = RGB, 1 = HSL, 2 = YUV, 3 = YIQ, 4 = XYZ
* \return true if everything is ok
*/
bool CxImage::Repair(float radius, long niterations, long colorspace)
{
if (!IsValid()) return false;
long w = GetWidth();
long h = GetHeight();
CxImage r,g,b;
r.Create(w,h,8);
g.Create(w,h,8);
b.Create(w,h,8);
switch (colorspace){
case 1:
SplitHSL(&r,&g,&b);
break;
case 2:
SplitYUV(&r,&g,&b);
break;
case 3:
SplitYIQ(&r,&g,&b);
break;
case 4:
SplitXYZ(&r,&g,&b);
break;
default:
SplitRGB(&r,&g,&b);
}
for (int i=0; i<niterations; i++){
RepairChannel(&r,radius);
RepairChannel(&g,radius);
RepairChannel(&b,radius);
}
CxImage* a=NULL;
#if CXIMAGE_SUPPORT_ALPHA
if (AlphaIsValid()){
a = new CxImage();
AlphaSplit(a);
}
#endif
Combine(&r,&g,&b,a,colorspace);
delete a;
return true;
}
////////////////////////////////////////////////////////////////////////////////
bool CxImage::RepairChannel(CxImage *ch, float radius)
{
if (ch==NULL) return false;
CxImage tmp(*ch);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long w = ch->GetWidth()-1;
long h = ch->GetHeight()-1;
double correction,ix,iy,ixx,ixy,iyy;
int x,y,xy0,xp1,xm1,yp1,ym1;
for(x=1; x<w; x++){
for(y=1; y<h; y++){
xy0 = ch->BlindGetPixelIndex(x,y);
xm1 = ch->BlindGetPixelIndex(x-1,y);
xp1 = ch->BlindGetPixelIndex(x+1,y);
ym1 = ch->BlindGetPixelIndex(x,y-1);
yp1 = ch->BlindGetPixelIndex(x,y+1);
ix= (xp1-xm1)/2.0;
iy= (yp1-ym1)/2.0;
ixx= xp1 - 2.0 * xy0 + xm1;
iyy= yp1 - 2.0 * xy0 + ym1;
ixy=(ch->BlindGetPixelIndex(x+1,y+1) + ch->BlindGetPixelIndex(x-1,y-1) -
ch->BlindGetPixelIndex(x-1,y+1) - ch->BlindGetPixelIndex(x+1,y-1))/4.0;
correction = ((1.0+iy*iy)*ixx - ix*iy*ixy + (1.0+ix*ix)*iyy)/(1.0+ix*ix+iy*iy);
tmp.BlindSetPixelIndex(x,y,(BYTE)min(255,max(0,(xy0 + radius * correction + 0.5))));
}
}
for (x=0;x<=w;x++){
for(y=0; y<=h; y+=h){
xy0 = ch->BlindGetPixelIndex(x,y);
xm1 = ch->GetPixelIndex(x-1,y);
xp1 = ch->GetPixelIndex(x+1,y);
ym1 = ch->GetPixelIndex(x,y-1);
yp1 = ch->GetPixelIndex(x,y+1);
ix= (xp1-xm1)/2.0;
iy= (yp1-ym1)/2.0;
ixx= xp1 - 2.0 * xy0 + xm1;
iyy= yp1 - 2.0 * xy0 + ym1;
ixy=(ch->GetPixelIndex(x+1,y+1) + ch->GetPixelIndex(x-1,y-1) -
ch->GetPixelIndex(x-1,y+1) - ch->GetPixelIndex(x+1,y-1))/4.0;
correction = ((1.0+iy*iy)*ixx - ix*iy*ixy + (1.0+ix*ix)*iyy)/(1.0+ix*ix+iy*iy);
tmp.BlindSetPixelIndex(x,y,(BYTE)min(255,max(0,(xy0 + radius * correction + 0.5))));
}
}
for (x=0;x<=w;x+=w){
for (y=0;y<=h;y++){
xy0 = ch->BlindGetPixelIndex(x,y);
xm1 = ch->GetPixelIndex(x-1,y);
xp1 = ch->GetPixelIndex(x+1,y);
ym1 = ch->GetPixelIndex(x,y-1);
yp1 = ch->GetPixelIndex(x,y+1);
ix= (xp1-xm1)/2.0;
iy= (yp1-ym1)/2.0;
ixx= xp1 - 2.0 * xy0 + xm1;
iyy= yp1 - 2.0 * xy0 + ym1;
ixy=(ch->GetPixelIndex(x+1,y+1) + ch->GetPixelIndex(x-1,y-1) -
ch->GetPixelIndex(x-1,y+1) - ch->GetPixelIndex(x+1,y-1))/4.0;
correction = ((1.0+iy*iy)*ixx - ix*iy*ixy + (1.0+ix*ix)*iyy)/(1.0+ix*ix+iy*iy);
tmp.BlindSetPixelIndex(x,y,(BYTE)min(255,max(0,(xy0 + radius * correction + 0.5))));
}
}
ch->Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Enhance the variations between adjacent pixels.
* Similar results can be achieved using Filter(),
* but the algorithms are different both in Edge() and in Contour().
* \return true if everything is ok
*/
bool CxImage::Contour()
{
if (!pDib) return false;
long Ksize = 3;
long k2 = Ksize/2;
long kmax= Ksize-k2;
long i,j,k;
BYTE maxr,maxg,maxb;
RGBQUAD pix1,pix2;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
pix1 = BlindGetPixelColor(x,y);
maxr=maxg=maxb=0;
for(j=-k2, i=0;j<kmax;j++){
for(k=-k2;k<kmax;k++, i++){
if (!IsInside(x+j,y+k)) continue;
pix2 = BlindGetPixelColor(x+j,y+k);
if ((pix2.rgbBlue-pix1.rgbBlue)>maxb) maxb = pix2.rgbBlue;
if ((pix2.rgbGreen-pix1.rgbGreen)>maxg) maxg = pix2.rgbGreen;
if ((pix2.rgbRed-pix1.rgbRed)>maxr) maxr = pix2.rgbRed;
}
}
pix1.rgbBlue=(BYTE)(255-maxb);
pix1.rgbGreen=(BYTE)(255-maxg);
pix1.rgbRed=(BYTE)(255-maxr);
tmp.BlindSetPixelColor(x,y,pix1);
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Adds a random offset to each pixel in the image
* \param radius: maximum pixel displacement
* \return true if everything is ok
*/
bool CxImage::Jitter(long radius)
{
if (!pDib) return false;
long nx,ny;
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
nx=x+(long)((rand()/(float)RAND_MAX - 0.5)*(radius*2));
ny=y+(long)((rand()/(float)RAND_MAX - 0.5)*(radius*2));
if (!IsInside(nx,ny)) {
nx=x;
ny=y;
}
if (head.biClrUsed==0){
tmp.BlindSetPixelColor(x,y,BlindGetPixelColor(nx,ny));
} else {
tmp.BlindSetPixelIndex(x,y,BlindGetPixelIndex(nx,ny));
}
#if CXIMAGE_SUPPORT_ALPHA
tmp.AlphaSet(x,y,AlphaGet(nx,ny));
#endif //CXIMAGE_SUPPORT_ALPHA
}
}
}
Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* generates a 1-D convolution matrix to be used for each pass of
* a two-pass gaussian blur. Returns the length of the matrix.
* \author [nipper]
*/
int CxImage::gen_convolve_matrix (float radius, float **cmatrix_p)
{
int matrix_length;
int matrix_midpoint;
float* cmatrix;
int i,j;
float std_dev;
float sum;
/* we want to generate a matrix that goes out a certain radius
* from the center, so we have to go out ceil(rad-0.5) pixels,
* inlcuding the center pixel. Of course, that's only in one direction,
* so we have to go the same amount in the other direction, but not count
* the center pixel again. So we double the previous result and subtract
* one.
* The radius parameter that is passed to this function is used as
* the standard deviation, and the radius of effect is the
* standard deviation * 2. It's a little confusing.
* <DP> modified scaling, so that matrix_lenght = 1+2*radius parameter
*/
radius = (float)fabs(0.5*radius) + 0.25f;
std_dev = radius;
radius = std_dev * 2;
/* go out 'radius' in each direction */
matrix_length = int (2 * ceil(radius-0.5) + 1);
if (matrix_length <= 0) matrix_length = 1;
matrix_midpoint = matrix_length/2 + 1;
*cmatrix_p = new float[matrix_length];
cmatrix = *cmatrix_p;
/* Now we fill the matrix by doing a numeric integration approximation
* from -2*std_dev to 2*std_dev, sampling 50 points per pixel.
* We do the bottom half, mirror it to the top half, then compute the
* center point. Otherwise asymmetric quantization errors will occur.
* The formula to integrate is e^-(x^2/2s^2).
*/
/* first we do the top (right) half of matrix */
for (i = matrix_length/2 + 1; i < matrix_length; i++)
{
float base_x = i - (float)floor((float)(matrix_length/2)) - 0.5f;
sum = 0;
for (j = 1; j <= 50; j++)
{
if ( base_x+0.02*j <= radius )
sum += (float)exp (-(base_x+0.02*j)*(base_x+0.02*j) /
(2*std_dev*std_dev));
}
cmatrix[i] = sum/50;
}
/* mirror the thing to the bottom half */
for (i=0; i<=matrix_length/2; i++) {
cmatrix[i] = cmatrix[matrix_length-1-i];
}
/* find center val -- calculate an odd number of quanta to make it symmetric,
* even if the center point is weighted slightly higher than others. */
sum = 0;
for (j=0; j<=50; j++)
{
sum += (float)exp (-(0.5+0.02*j)*(0.5+0.02*j) /
(2*std_dev*std_dev));
}
cmatrix[matrix_length/2] = sum/51;
/* normalize the distribution by scaling the total sum to one */
sum=0;
for (i=0; i<matrix_length; i++) sum += cmatrix[i];
for (i=0; i<matrix_length; i++) cmatrix[i] = cmatrix[i] / sum;
return matrix_length;
}
////////////////////////////////////////////////////////////////////////////////
/**
* generates a lookup table for every possible product of 0-255 and
* each value in the convolution matrix. The returned array is
* indexed first by matrix position, then by input multiplicand (?)
* value.
* \author [nipper]
*/
float* CxImage::gen_lookup_table (float *cmatrix, int cmatrix_length)
{
float* lookup_table = new float[cmatrix_length * 256];
float* lookup_table_p = lookup_table;
float* cmatrix_p = cmatrix;
for (int i=0; i<cmatrix_length; i++)
{
for (int j=0; j<256; j++)
{
*(lookup_table_p++) = *cmatrix_p * (float)j;
}
cmatrix_p++;
}
return lookup_table;
}
////////////////////////////////////////////////////////////////////////////////
/**
* this function is written as if it is blurring a column at a time,
* even though it can operate on rows, too. There is no difference
* in the processing of the lines, at least to the blur_line function.
* \author [nipper]
*/
void CxImage::blur_line (float *ctable, float *cmatrix, int cmatrix_length, BYTE* cur_col, BYTE* dest_col, int y, long bytes)
{
float scale;
float sum;
int i=0, j=0;
int row;
int cmatrix_middle = cmatrix_length/2;
float *cmatrix_p;
BYTE *cur_col_p;
BYTE *cur_col_p1;
BYTE *dest_col_p;
float *ctable_p;
/* this first block is the same as the non-optimized version --
* it is only used for very small pictures, so speed isn't a
* big concern.
*/
if (cmatrix_length > y)
{
for (row = 0; row < y ; row++)
{
scale=0;
/* find the scale factor */
for (j = 0; j < y ; j++)
{
/* if the index is in bounds, add it to the scale counter */
if ((j + cmatrix_middle - row >= 0) &&
(j + cmatrix_middle - row < cmatrix_length))
scale += cmatrix[j + cmatrix_middle - row];
}
for (i = 0; i<bytes; i++)
{
sum = 0;
for (j = 0; j < y; j++)
{
if ((j >= row - cmatrix_middle) &&
(j <= row + cmatrix_middle))
sum += cur_col[j*bytes + i] * cmatrix[j];
}
dest_col[row*bytes + i] = (BYTE)(0.5f + sum / scale);
}
}
}
else
{
/* for the edge condition, we only use available info and scale to one */
for (row = 0; row < cmatrix_middle; row++)
{
/* find scale factor */
scale=0;
for (j = cmatrix_middle - row; j<cmatrix_length; j++)
scale += cmatrix[j];
for (i = 0; i<bytes; i++)
{
sum = 0;
for (j = cmatrix_middle - row; j<cmatrix_length; j++)
{
sum += cur_col[(row + j-cmatrix_middle)*bytes + i] * cmatrix[j];
}
dest_col[row*bytes + i] = (BYTE)(0.5f + sum / scale);
}
}
/* go through each pixel in each col */
dest_col_p = dest_col + row*bytes;
for (; row < y-cmatrix_middle; row++)
{
cur_col_p = (row - cmatrix_middle) * bytes + cur_col;
for (i = 0; i<bytes; i++)
{
sum = 0;
cmatrix_p = cmatrix;
cur_col_p1 = cur_col_p;
ctable_p = ctable;
for (j = cmatrix_length; j>0; j--)
{
sum += *(ctable_p + *cur_col_p1);
cur_col_p1 += bytes;
ctable_p += 256;
}
cur_col_p++;
*(dest_col_p++) = (BYTE)(0.5f + sum);
}
}
/* for the edge condition , we only use available info, and scale to one */
for (; row < y; row++)
{
/* find scale factor */
scale=0;
for (j = 0; j< y-row + cmatrix_middle; j++)
scale += cmatrix[j];
for (i = 0; i<bytes; i++)
{
sum = 0;
for (j = 0; j<y-row + cmatrix_middle; j++)
{
sum += cur_col[(row + j-cmatrix_middle)*bytes + i] * cmatrix[j];
}
dest_col[row*bytes + i] = (BYTE) (0.5f + sum / scale);
}
}
}
}
////////////////////////////////////////////////////////////////////////////////
/**
* \author [DP]
*/
void CxImage::blur_text (BYTE threshold, BYTE decay, BYTE max_depth, CxImage* iSrc, CxImage* iDst, BYTE bytes)
{
long x,y,z,m;
BYTE *pSrc, *pSrc2, *pSrc3, *pDst;
BYTE step,n;
int pivot;
if (max_depth<1) max_depth = 1;
long nmin,nmax,xmin,xmax,ymin,ymax;
xmin = ymin = 0;
xmax = iSrc->head.biWidth;
ymax = iSrc->head.biHeight;
if (xmin==xmax || ymin==ymax) return;
nmin = xmin * bytes;
nmax = xmax * bytes;
CImageIterator itSrc(iSrc);
CImageIterator itTmp(iDst);
double dbScaler = 100.0f/(ymax-ymin)/bytes;
for (n=0; n<bytes; n++){
for (y=ymin+1;y<(ymax-1);y++)
{
if (info.nEscape) break;
info.nProgress = (long)((y-ymin)*dbScaler*(1+n));
pSrc = itSrc.GetRow(y);
pSrc2 = itSrc.GetRow(y+1);
pSrc3 = itSrc.GetRow(y-1);
pDst = itTmp.GetRow(y);
//scan left to right
for (x=n+nmin /*,i=xmin*/; x<(nmax-1); x+=bytes /*,i++*/)
{
z=x+bytes;
pivot = pSrc[z]-threshold;
//find upper corner
if (pSrc[x]<pivot && pSrc2[z]<pivot && pSrc3[x]>=pivot){
while (z<nmax && pSrc2[z]<pSrc[x+bytes] && pSrc[x+bytes]<=pSrc[z]){
z+=bytes;
}
m = z-x;
m = (decay>1) ? ((m/bytes)/decay+1) : m/bytes;
if (m>max_depth) m = max_depth;
step = (BYTE)((pSrc[x+bytes]-pSrc[x])/(m+1));
while (m-->1){
pDst[x+m*bytes] = (BYTE)(pDst[x]+(step*(m+1)));
}
}
//find lower corner
z=x+bytes;
if (pSrc[x]<pivot && pSrc3[z]<pivot && pSrc2[x]>=pivot){
while (z<nmax && pSrc3[z]<pSrc[x+bytes] && pSrc[x+bytes]<=pSrc[z]){
z+=bytes;
}
m = z-x;
m = (decay>1) ? ((m/bytes)/decay+1) : m/bytes;
if (m>max_depth) m = max_depth;
step = (BYTE)((pSrc[x+bytes]-pSrc[x])/(m+1));
while (m-->1){
pDst[x+m*bytes] = (BYTE)(pDst[x]+(step*(m+1)));
}
}
}
//scan right to left
for (x=nmax-1-n /*,i=(xmax-1)*/; x>0; x-=bytes /*,i--*/)
{
z=x-bytes;
pivot = pSrc[z]-threshold;
//find upper corner
if (pSrc[x]<pivot && pSrc2[z]<pivot && pSrc3[x]>=pivot){
while (z>n && pSrc2[z]<pSrc[x-bytes] && pSrc[x-bytes]<=pSrc[z]){
z-=bytes;
}
m = x-z;
m = (decay>1) ? ((m/bytes)/decay+1) : m/bytes;
if (m>max_depth) m = max_depth;
step = (BYTE)((pSrc[x-bytes]-pSrc[x])/(m+1));
while (m-->1){
pDst[x-m*bytes] = (BYTE)(pDst[x]+(step*(m+1)));
}
}
//find lower corner
z=x-bytes;
if (pSrc[x]<pivot && pSrc3[z]<pivot && pSrc2[x]>=pivot){
while (z>n && pSrc3[z]<pSrc[x-bytes] && pSrc[x-bytes]<=pSrc[z]){
z-=bytes;
}
m = x-z;
m = (decay>1) ? ((m/bytes)/decay+1) : m/bytes;
if (m>max_depth) m = max_depth;
step = (BYTE)((pSrc[x-bytes]-pSrc[x])/(m+1));
while (m-->1){
pDst[x-m*bytes] = (BYTE)(pDst[x]+(step*(m+1)));
}
}
}
}
}
}
////////////////////////////////////////////////////////////////////////////////
/**
* \author [DP]
*/
bool CxImage::TextBlur(BYTE threshold, BYTE decay, BYTE max_depth, bool bBlurHorizontal, bool bBlurVertical, CxImage* iDst)
{
if (!pDib) return false;
RGBQUAD* pPalette=NULL;
WORD bpp = GetBpp();
//the routine is optimized for RGB or GrayScale images
if (!(head.biBitCount == 24 || IsGrayScale())){
pPalette = new RGBQUAD[head.biClrUsed];
memcpy(pPalette, GetPalette(),GetPaletteSize());
if (!IncreaseBpp(24))
return false;
}
CxImage tmp(*this);
if (!tmp.IsValid()){
strcpy(info.szLastError,tmp.GetLastError());
return false;
}
if (bBlurHorizontal)
blur_text(threshold, decay, max_depth, this, &tmp, head.biBitCount>>3);
if (bBlurVertical){
CxImage src2(*this);
src2.RotateLeft();
tmp.RotateLeft();
blur_text(threshold, decay, max_depth, &src2, &tmp, head.biBitCount>>3);
tmp.RotateRight();
}
#if CXIMAGE_SUPPORT_SELECTION
//restore the non selected region
if (pSelection){
for(long y=0; y<head.biHeight; y++){
for(long x=0; x<head.biWidth; x++){
if (!BlindSelectionIsInside(x,y)){
tmp.BlindSetPixelColor(x,y,BlindGetPixelColor(x,y));
}
}
}
}
#endif //CXIMAGE_SUPPORT_SELECTION
//if necessary, restore the original BPP and palette
if (pPalette){
tmp.DecreaseBpp(bpp, true, pPalette);
delete [] pPalette;
}
if (iDst) iDst->Transfer(tmp);
else Transfer(tmp);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* \author [nipper]; changes [DP]
*/
bool CxImage::GaussianBlur(float radius /*= 1.0f*/, CxImage* iDst /*= 0*/)
{
if (!pDib) return false;
RGBQUAD* pPalette=NULL;
WORD bpp = GetBpp();
//the routine is optimized for RGB or GrayScale images
if (!(head.biBitCount == 24 || IsGrayScale())){
pPalette = new RGBQUAD[head.biClrUsed];
memcpy(pPalette, GetPalette(),GetPaletteSize());
if (!IncreaseBpp(24))
return false;
}
CxImage tmp_x(*this, false, true, true);
if (!tmp_x.IsValid()){
strcpy(info.szLastError,tmp_x.GetLastError());
return false;
}
// generate convolution matrix and make sure it's smaller than each dimension
float *cmatrix = NULL;
int cmatrix_length = gen_convolve_matrix(radius, &cmatrix);
// generate lookup table
float *ctable = gen_lookup_table(cmatrix, cmatrix_length);
long x,y;
int bypp = head.biBitCount>>3;
CImageIterator itSrc(this);
CImageIterator itTmp(&tmp_x);
double dbScaler = 50.0f/head.biHeight;
// blur the rows
for (y=0;y<head.biHeight;y++)
{
if (info.nEscape) break;
info.nProgress = (long)(y*dbScaler);
blur_line(ctable, cmatrix, cmatrix_length, itSrc.GetRow(y), itTmp.GetRow(y), head.biWidth, bypp);
}
CxImage tmp_y(tmp_x, false, true, true);
if (!tmp_y.IsValid()){
strcpy(info.szLastError,tmp_y.GetLastError());
return false;
}
CImageIterator itDst(&tmp_y);
// blur the cols
BYTE* cur_col = (BYTE*)malloc(bypp*head.biHeight);
BYTE* dest_col = (BYTE*)malloc(bypp*head.biHeight);
dbScaler = 50.0f/head.biWidth;
for (x=0;x<head.biWidth;x++)
{
if (info.nEscape) break;
info.nProgress = (long)(50.0f+x*dbScaler);
itTmp.GetCol(cur_col, x);
itDst.GetCol(dest_col, x);
blur_line(ctable, cmatrix, cmatrix_length, cur_col, dest_col, head.biHeight, bypp);
itDst.SetCol(dest_col, x);
}
free(cur_col);
free(dest_col);
delete [] cmatrix;
delete [] ctable;
#if CXIMAGE_SUPPORT_SELECTION
//restore the non selected region
if (pSelection){
for(y=0; y<head.biHeight; y++){
for(x=0; x<head.biWidth; x++){
if (!BlindSelectionIsInside(x,y)){
tmp_y.BlindSetPixelColor(x,y,BlindGetPixelColor(x,y));
}
}
}
}
#endif //CXIMAGE_SUPPORT_SELECTION
//if necessary, restore the original BPP and palette
if (pPalette){
tmp_y.DecreaseBpp(bpp, false, pPalette);
if (iDst) DecreaseBpp(bpp, false, pPalette);
delete [] pPalette;
}
if (iDst) iDst->Transfer(tmp_y);
else Transfer(tmp_y);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* \author [DP],[nipper]
*/
bool CxImage::SelectiveBlur(float radius, BYTE threshold, CxImage* iDst)
{
if (!pDib) return false;
RGBQUAD* pPalette=NULL;
WORD bpp = GetBpp();
CxImage Tmp(*this, true, true, true);
if (!Tmp.IsValid()){
strcpy(info.szLastError,Tmp.GetLastError());
return false;
}
//the routine is optimized for RGB or GrayScale images
if (!(head.biBitCount == 24 || IsGrayScale())){
pPalette = new RGBQUAD[head.biClrUsed];
memcpy(pPalette, GetPalette(),GetPaletteSize());
if (!Tmp.IncreaseBpp(24))
return false;
}
CxImage Dst(Tmp, true, true, true);
if (!Dst.IsValid()){
strcpy(info.szLastError,Dst.GetLastError());
return false;
}
//build the difference mask
BYTE thresh_dw = (BYTE)max( 0 ,(int)(128 - threshold));
BYTE thresh_up = (BYTE)min(255,(int)(128 + threshold));
long kernel[]={-100,-100,-100,-100,801,-100,-100,-100,-100};
if (!Tmp.Filter(kernel,3,800,128)){
strcpy(info.szLastError,Tmp.GetLastError());
return false;
}
//if the image has no selection, build a selection for the whole image
if (!Tmp.SelectionIsValid()){
Tmp.SelectionCreate();
Tmp.SelectionClear(255);
}
long xmin,xmax,ymin,ymax;
xmin = Tmp.info.rSelectionBox.left;
xmax = Tmp.info.rSelectionBox.right;
ymin = Tmp.info.rSelectionBox.bottom;
ymax = Tmp.info.rSelectionBox.top;
//modify the selection where the difference mask is over the threshold
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
if(Tmp.BlindSelectionIsInside(x,y)){
RGBQUAD c = Tmp.BlindGetPixelColor(x,y);
if ((c.rgbRed < thresh_dw || c.rgbRed > thresh_up) ||
(c.rgbGreen < thresh_dw || c.rgbGreen > thresh_up) ||
(c.rgbBlue < thresh_dw || c.rgbBlue > thresh_up))
{
Tmp.SelectionSet(x,y,0);
}
}
}
}
//blur the image (only in the selected pixels)
Dst.SelectionCopy(Tmp);
if (!Dst.GaussianBlur(radius)){
strcpy(info.szLastError,Dst.GetLastError());
return false;
}
//restore the original selection
Dst.SelectionCopy(*this);
//if necessary, restore the original BPP and palette
if (pPalette){
Dst.DecreaseBpp(bpp, false, pPalette);
delete [] pPalette;
}
if (iDst) iDst->Transfer(Dst);
else Transfer(Dst);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* sharpen the image by subtracting a blurred copy from the original image.
* \param radius: width in pixels of the blurring effect. Range: >0; default = 5.
* \param amount: strength of the filter. Range: 0.0 (none) to 1.0 (max); default = 0.5
* \param threshold: difference, between blurred and original pixel, to trigger the filter
* Range: 0 (always triggered) to 255 (never triggered); default = 0.
* \return true if everything is ok
* \author [nipper]; changes [DP]
*/
bool CxImage::UnsharpMask(float radius /*= 5.0*/, float amount /*= 0.5*/, int threshold /*= 0*/)
{
if (!pDib) return false;
RGBQUAD* pPalette=NULL;
WORD bpp = GetBpp();
//the routine is optimized for RGB or GrayScale images
if (!(head.biBitCount == 24 || IsGrayScale())){
pPalette = new RGBQUAD[head.biClrUsed];
memcpy(pPalette, GetPalette(),GetPaletteSize());
if (!IncreaseBpp(24))
return false;
}
CxImage iDst;
if (!GaussianBlur(radius,&iDst))
return false;
CImageIterator itSrc(this);
CImageIterator itDst(&iDst);
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin==xmax || ymin==ymax)
return false;
double dbScaler = 100.0/(ymax-ymin);
int bypp = head.biBitCount>>3;
// merge the source and destination (which currently contains
// the blurred version) images
for (long y=ymin; y<ymax; y++)
{
if (info.nEscape) break;
info.nProgress = (long)((y-ymin)*dbScaler);
// get source row
BYTE* cur_row = itSrc.GetRow(y);
// get dest row
BYTE* dest_row = itDst.GetRow(y);
// combine the two
for (long x=xmin; x<xmax; x++) {
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
for (long b=0, z=x*bypp; b<bypp; b++, z++){
int diff = cur_row[z] - dest_row[z];
// do tresholding
if (abs(diff) < threshold){
dest_row[z] = cur_row[z];
} else {
dest_row[z] = (BYTE)min(255, max(0,(int)(cur_row[z] + amount * diff)));
}
}
}
}
}
//if necessary, restore the original BPP and palette
if (pPalette){
iDst.DecreaseBpp(bpp, false, pPalette);
delete [] pPalette;
}
Transfer(iDst);
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Apply a look up table to the image.
* \param pLut: BYTE[256] look up table
* \return true if everything is ok
*/
bool CxImage::Lut(BYTE* pLut)
{
if (!pDib || !pLut) return false;
RGBQUAD color;
double dbScaler;
if (head.biClrUsed==0){
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
// faster loop for full image
BYTE *iSrc=info.pImage;
for(unsigned long i=0; i < head.biSizeImage ; i++){
*iSrc++ = pLut[*iSrc];
}
return true;
}
if (xmin==xmax || ymin==ymax)
return false;
dbScaler = 100.0/(ymax-ymin);
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)((y-ymin)*dbScaler); //<Anatoly Ivasyuk>
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
color = BlindGetPixelColor(x,y);
color.rgbRed = pLut[color.rgbRed];
color.rgbGreen = pLut[color.rgbGreen];
color.rgbBlue = pLut[color.rgbBlue];
BlindSetPixelColor(x,y,color);
}
}
}
#if CXIMAGE_SUPPORT_SELECTION
} else if (pSelection && (head.biBitCount==8) && IsGrayScale()){
long xmin,xmax,ymin,ymax;
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
if (xmin==xmax || ymin==ymax)
return false;
dbScaler = 100.0/(ymax-ymin);
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)((y-ymin)*dbScaler);
for(long x=xmin; x<xmax; x++){
if (BlindSelectionIsInside(x,y))
{
BlindSetPixelIndex(x,y,pLut[BlindGetPixelIndex(x,y)]);
}
}
}
#endif //CXIMAGE_SUPPORT_SELECTION
} else {
bool bIsGrayScale = IsGrayScale();
for(DWORD j=0; j<head.biClrUsed; j++){
color = GetPaletteColor((BYTE)j);
color.rgbRed = pLut[color.rgbRed];
color.rgbGreen = pLut[color.rgbGreen];
color.rgbBlue = pLut[color.rgbBlue];
SetPaletteColor((BYTE)j,color);
}
if (bIsGrayScale) GrayScale();
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Apply an indipendent look up table for each channel
* \param pLutR, pLutG, pLutB, pLutA: BYTE[256] look up tables
* \return true if everything is ok
*/
bool CxImage::Lut(BYTE* pLutR, BYTE* pLutG, BYTE* pLutB, BYTE* pLutA)
{
if (!pDib || !pLutR || !pLutG || !pLutB) return false;
RGBQUAD color;
double dbScaler;
if (head.biClrUsed==0){
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin==xmax || ymin==ymax)
return false;
dbScaler = 100.0/(ymax-ymin);
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)((y-ymin)*dbScaler);
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
color = BlindGetPixelColor(x,y);
color.rgbRed = pLutR[color.rgbRed];
color.rgbGreen = pLutG[color.rgbGreen];
color.rgbBlue = pLutB[color.rgbBlue];
if (pLutA) color.rgbReserved=pLutA[color.rgbReserved];
BlindSetPixelColor(x,y,color,true);
}
}
}
} else {
bool bIsGrayScale = IsGrayScale();
for(DWORD j=0; j<head.biClrUsed; j++){
color = GetPaletteColor((BYTE)j);
color.rgbRed = pLutR[color.rgbRed];
color.rgbGreen = pLutG[color.rgbGreen];
color.rgbBlue = pLutB[color.rgbBlue];
SetPaletteColor((BYTE)j,color);
}
if (bIsGrayScale) GrayScale();
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Use the RedEyeRemove function to remove the red-eye effect that frequently
* occurs in photographs of humans and animals. You must select the region
* where the function will filter the red channel.
* \param strength: range from 0.0f (no effect) to 1.0f (full effect). Default = 0.8
* \return true if everything is ok
*/
bool CxImage::RedEyeRemove(float strength)
{
if (!pDib) return false;
RGBQUAD color;
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin==xmax || ymin==ymax)
return false;
if (strength<0.0f) strength = 0.0f;
if (strength>1.0f) strength = 1.0f;
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
float a = 1.0f-5.0f*((float)((x-0.5f*(xmax+xmin))*(x-0.5f*(xmax+xmin))+(y-0.5f*(ymax+ymin))*(y-0.5f*(ymax+ymin))))/((float)((xmax-xmin)*(ymax-ymin)));
if (a<0) a=0;
color = BlindGetPixelColor(x,y);
color.rgbRed = (BYTE)(a*min(color.rgbGreen,color.rgbBlue)+(1.0f-a)*color.rgbRed);
BlindSetPixelColor(x,y,color);
}
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Changes the saturation of the image.
* \param saturation: can be from -100 to 100, positive values increase the saturation.
* \param colorspace: can be 1 (HSL) or 2 (YUV).
* \return true if everything is ok
*/
bool CxImage::Saturate(const long saturation, const long colorspace)
{
if (!pDib)
return false;
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin==xmax || ymin==ymax)
return false;
BYTE cTable[256];
switch(colorspace)
{
case 1:
{
for (int i=0;i<256;i++) {
cTable[i] = (BYTE)max(0,min(255,(int)(i + saturation)));
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
RGBQUAD c = RGBtoHSL(BlindGetPixelColor(x,y));
c.rgbGreen = cTable[c.rgbGreen];
c = HSLtoRGB(c);
BlindSetPixelColor(x,y,c);
}
}
}
}
break;
case 2:
{
for (int i=0;i<256;i++) {
cTable[i] = (BYTE)max(0,min(255,(int)((i-128)*(100 + saturation)/100.0f + 128.5f)));
}
for(long y=ymin; y<ymax; y++){
info.nProgress = (long)(100*(y-ymin)/(ymax-ymin));
if (info.nEscape) break;
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
RGBQUAD c = RGBtoYUV(BlindGetPixelColor(x,y));
c.rgbGreen = cTable[c.rgbGreen];
c.rgbBlue = cTable[c.rgbBlue];
c = YUVtoRGB(c);
BlindSetPixelColor(x,y,c);
}
}
}
}
break;
default:
strcpy(info.szLastError,"Saturate: wrong colorspace");
return false;
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Solarize: convert all colors above a given lightness level into their negative
* \param level : lightness threshold. Range = 0 to 255; default = 128.
* \param bLinkedChannels: true = compare with luminance, preserve colors (default)
* false = compare with independent R,G,B levels
* \return true if everything is ok
* \author [Priyank Bolia] (priyank_bolia(at)yahoo(dot)com); changes [DP]
*/
bool CxImage::Solarize(BYTE level, bool bLinkedChannels)
{
if (!pDib) return false;
long xmin,xmax,ymin,ymax;
if (pSelection){
xmin = info.rSelectionBox.left; xmax = info.rSelectionBox.right;
ymin = info.rSelectionBox.bottom; ymax = info.rSelectionBox.top;
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (head.biBitCount<=8){
if (IsGrayScale()){ //GRAYSCALE, selection
for(long y=ymin; y<ymax; y++){
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
BYTE index = BlindGetPixelIndex(x,y);
RGBQUAD color = GetPaletteColor(index);
if ((BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue)>level){
BlindSetPixelIndex(x,y,255-index);
}
}
}
}
} else { //PALETTE, full image
RGBQUAD* ppal=GetPalette();
for(DWORD i=0;i<head.biClrUsed;i++){
RGBQUAD color = GetPaletteColor((BYTE)i);
if (bLinkedChannels){
if ((BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue)>level){
ppal[i].rgbBlue =(BYTE)(255-ppal[i].rgbBlue);
ppal[i].rgbGreen =(BYTE)(255-ppal[i].rgbGreen);
ppal[i].rgbRed =(BYTE)(255-ppal[i].rgbRed);
}
} else {
if (color.rgbBlue>level) ppal[i].rgbBlue =(BYTE)(255-ppal[i].rgbBlue);
if (color.rgbGreen>level) ppal[i].rgbGreen =(BYTE)(255-ppal[i].rgbGreen);
if (color.rgbRed>level) ppal[i].rgbRed =(BYTE)(255-ppal[i].rgbRed);
}
}
}
} else { //RGB, selection
for(long y=ymin; y<ymax; y++){
for(long x=xmin; x<xmax; x++){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
RGBQUAD color = BlindGetPixelColor(x,y);
if (bLinkedChannels){
if ((BYTE)RGB2GRAY(color.rgbRed,color.rgbGreen,color.rgbBlue)>level){
color.rgbRed = (BYTE)(255-color.rgbRed);
color.rgbGreen = (BYTE)(255-color.rgbGreen);
color.rgbBlue = (BYTE)(255-color.rgbBlue);
}
} else {
if (color.rgbBlue>level) color.rgbBlue =(BYTE)(255-color.rgbBlue);
if (color.rgbGreen>level) color.rgbGreen =(BYTE)(255-color.rgbGreen);
if (color.rgbRed>level) color.rgbRed =(BYTE)(255-color.rgbRed);
}
BlindSetPixelColor(x,y,color);
}
}
}
}
//invert transparent color only in case of full image processing
if (pSelection==0 || (!IsGrayScale() && IsIndexed())){
if (bLinkedChannels){
if ((BYTE)RGB2GRAY(info.nBkgndColor.rgbRed,info.nBkgndColor.rgbGreen,info.nBkgndColor.rgbBlue)>level){
info.nBkgndColor.rgbBlue = (BYTE)(255-info.nBkgndColor.rgbBlue);
info.nBkgndColor.rgbGreen = (BYTE)(255-info.nBkgndColor.rgbGreen);
info.nBkgndColor.rgbRed = (BYTE)(255-info.nBkgndColor.rgbRed);
}
} else {
if (info.nBkgndColor.rgbBlue>level) info.nBkgndColor.rgbBlue = (BYTE)(255-info.nBkgndColor.rgbBlue);
if (info.nBkgndColor.rgbGreen>level) info.nBkgndColor.rgbGreen = (BYTE)(255-info.nBkgndColor.rgbGreen);
if (info.nBkgndColor.rgbRed>level) info.nBkgndColor.rgbRed = (BYTE)(255-info.nBkgndColor.rgbRed);
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Converts the RGB triplets to and from different colorspace
* \param dstColorSpace: destination colorspace; 0 = RGB, 1 = HSL, 2 = YUV, 3 = YIQ, 4 = XYZ
* \param srcColorSpace: source colorspace; 0 = RGB, 1 = HSL, 2 = YUV, 3 = YIQ, 4 = XYZ
* \return true if everything is ok
*/
bool CxImage::ConvertColorSpace(const long dstColorSpace, const long srcColorSpace)
{
if (!pDib)
return false;
if (dstColorSpace == srcColorSpace)
return true;
long w = GetWidth();
long h = GetHeight();
for (long y=0;y<h;y++){
info.nProgress = (long)(100*y/h);
if (info.nEscape) break;
for (long x=0;x<w;x++){
RGBQUAD c = BlindGetPixelColor(x,y);
switch (srcColorSpace){
case 0:
break;
case 1:
c = HSLtoRGB(c);
break;
case 2:
c = YUVtoRGB(c);
break;
case 3:
c = YIQtoRGB(c);
break;
case 4:
c = XYZtoRGB(c);
break;
default:
strcpy(info.szLastError,"ConvertColorSpace: unknown source colorspace");
return false;
}
switch (dstColorSpace){
case 0:
break;
case 1:
c = RGBtoHSL(c);
break;
case 2:
c = RGBtoYUV(c);
break;
case 3:
c = RGBtoYIQ(c);
break;
case 4:
c = RGBtoXYZ(c);
break;
default:
strcpy(info.szLastError,"ConvertColorSpace: unknown destination colorspace");
return false;
}
BlindSetPixelColor(x,y,c);
}
}
return true;
}
////////////////////////////////////////////////////////////////////////////////
/**
* Finds the optimal (global or local) treshold for image binarization
* \param method: 0 = average all methods (default); 1 = Otsu; 2 = Kittler & Illingworth; 3 = max entropy; 4 = potential difference;
* \param pBox: region from where the threshold is computed; 0 = full image (default).
* \param pContrastMask: limit the computation only in regions with contrasted (!=0) pixels; default = 0.
* the pContrastMask image must be grayscale with same with and height of the current image,
* can be obtained from the current image with a filter:
* CxImage iContrastMask(*image,true,false,false);
* iContrastMask.GrayScale();
* long edge[]={-1,-1,-1,-1,8,-1,-1,-1,-1};
* iContrastMask.Filter(edge,3,1,0);
* long blur[]={1,1,1,1,1,1,1,1,1};
* iContrastMask.Filter(blur,3,9,0);
* \return optimal threshold; -1 = error.
* \sa AdaptiveThreshold
*/
int CxImage::OptimalThreshold(long method, RECT * pBox, CxImage* pContrastMask)
{
if (!pDib)
return false;
if (head.biBitCount!=8){
strcpy(info.szLastError,"OptimalThreshold works only on 8 bit images");
return -1;
}
if (pContrastMask){
if (!pContrastMask->IsValid() ||
!pContrastMask->IsGrayScale() ||
pContrastMask->GetWidth() != GetWidth() ||
pContrastMask->GetHeight() != GetHeight()){
strcpy(info.szLastError,"OptimalThreshold invalid ContrastMask");
return -1;
}
}
long xmin,xmax,ymin,ymax;
if (pBox){
xmin = max(pBox->left,0);
xmax = min(pBox->right,head.biWidth);
ymin = max(pBox->bottom,0);
ymax = min(pBox->top,head.biHeight);
} else {
xmin = ymin = 0;
xmax = head.biWidth; ymax=head.biHeight;
}
if (xmin>=xmax || ymin>=ymax)
return -1;
double p[256];
memset(p, 0, 256*sizeof(double));
//build histogram
for (long y = ymin; y<ymax; y++){
BYTE* pGray = GetBits(y) + xmin;
BYTE* pContr = 0;
if (pContrastMask) pContr = pContrastMask->GetBits(y) + xmin;
for (long x = xmin; x<xmax; x++){
BYTE n = *pGray++;
if (pContr){
if (*pContr) p[n]++;
pContr++;
} else {
p[n]++;
}
}
}
//find histogram limits
int gray_min = 0;
while (gray_min<255 && p[gray_min]==0) gray_min++;
int gray_max = 255;
while (gray_max>0 && p[gray_max]==0) gray_max--;
if (gray_min > gray_max)
return -1;
if (gray_min == gray_max){
if (gray_min == 0)
return 0;
else
return gray_max-1;
}
//compute total moments 0th,1st,2nd order
int i,k;
double w_tot = 0;
double m_tot = 0;
double q_tot = 0;
for (i = gray_min; i <= gray_max; i++){
w_tot += p[i];
m_tot += i*p[i];
q_tot += i*i*p[i];
}
double L, L1max, L2max, L3max, L4max; //objective functions
int th1,th2,th3,th4; //optimal thresholds
L1max = L2max = L3max = L4max = 0;
th1 = th2 = th3 = th4 = -1;
double w1, w2, m1, m2, q1, q2, s1, s2;
w1 = m1 = q1 = 0;
for (i = gray_min; i < gray_max; i++){
w1 += p[i];
w2 = w_tot - w1;
m1 += i*p[i];
m2 = m_tot - m1;
q1 += i*i*p[i];
q2 = q_tot - q1;
s1 = q1/w1-m1*m1/w1/w1; //s1 = q1/w1-pow(m1/w1,2);
s2 = q2/w2-m2*m2/w2/w2; //s2 = q2/w2-pow(m2/w2,2);
//Otsu
L = -(s1*w1 + s2*w2); //implemented as definition
//L = w1 * w2 * (m2/w2 - m1/w1)*(m2/w2 - m1/w1); //implementation that doesn't need s1 & s2
if (L1max < L || th1<0){
L1max = L;
th1 = i;
}
//Kittler and Illingworth
if (s1>0 && s2>0){
L = w1*log(w1/sqrt(s1))+w2*log(w2/sqrt(s2));
//L = w1*log(w1*w1/s1)+w2*log(w2*w2/s2);
if (L2max < L || th2<0){
L2max = L;
th2 = i;
}
}
//max entropy
L = 0;
for (k=gray_min;k<=i;k++) if (p[k] > 0) L -= p[k]*log(p[k]/w1)/w1;
for (k;k<=gray_max;k++) if (p[k] > 0) L -= p[k]*log(p[k]/w2)/w2;
if (L3max < L || th3<0){
L3max = L;
th3 = i;
}
//potential difference (based on Electrostatic Binarization method by J. Acharya & G. Sreechakra)
// L=-fabs(vdiff/vsum); ? molto selettivo, sembra che L=-fabs(vdiff) o L=-(vsum)
// abbiano lo stesso valore di soglia... il che semplificherebbe molto la routine
double vdiff = 0;
for (k=gray_min;k<=i;k++)
vdiff += p[k]*(i-k)*(i-k);
double vsum = vdiff;
for (k;k<=gray_max;k++){
double dv = p[k]*(k-i)*(k-i);
vdiff -= dv;
vsum += dv;
}
if (vsum>0) L = -fabs(vdiff/vsum); else L = 0;
if (L4max < L || th4<0){
L4max = L;
th4 = i;
}
}
int threshold;
switch (method){
case 1: //Otsu
threshold = th1;
break;
case 2: //Kittler and Illingworth
threshold = th2;
break;
case 3: //max entropy
threshold = th3;
break;
case 4: //potential difference
threshold = th4;
break;
default: //auto
{
int nt = 0;
threshold = 0;
if (th1>=0) { threshold += th1; nt++;}
if (th2>=0) { threshold += th2; nt++;}
if (th3>=0) { threshold += th3; nt++;}
if (th4>=0) { threshold += th4; nt++;}
if (nt)
threshold /= nt;
else
threshold = (gray_min+gray_max)/2;
/*better(?) but really expensive alternative:
n = 0:255;
pth1 = c1(th1)/sqrt(2*pi*s1(th1))*exp(-((n - m1(th1)).^2)/2/s1(th1)) + c2(th1)/sqrt(2*pi*s2(th1))*exp(-((n - m2(th1)).^2)/2/s2(th1));
pth2 = c1(th2)/sqrt(2*pi*s1(th2))*exp(-((n - m1(th2)).^2)/2/s1(th2)) + c2(th2)/sqrt(2*pi*s2(th2))*exp(-((n - m2(th2)).^2)/2/s2(th2));
...
mse_th1 = sum((p-pth1).^2);
mse_th2 = sum((p-pth2).^2);
...
select th# that gives minimum mse_th#
*/
}
}
if (threshold <= gray_min || threshold >= gray_max)
threshold = (gray_min+gray_max)/2;
return threshold;
}
///////////////////////////////////////////////////////////////////////////////
/**
* Converts the image to B&W, using an optimal threshold mask
* \param method: 0 = average all methods (default); 1 = Otsu; 2 = Kittler & Illingworth; 3 = max entropy; 4 = potential difference;
* \param nBoxSize: the image is divided into "nBoxSize x nBoxSize" blocks, from where the threshold is computed; min = 8; default = 64.
* \param pContrastMask: limit the computation only in regions with contrasted (!=0) pixels; default = 0.
* \param nBias: global offset added to the threshold mask; default = 0.
* \param fGlobalLocalBalance: balance between local and global threshold. default = 0.5
* fGlobalLocalBalance can be from 0.0 (use only local threshold) to 1.0 (use only global threshold)
* the pContrastMask image must be grayscale with same with and height of the current image,
* \return true if everything is ok.
* \sa OptimalThreshold
*/
bool CxImage::AdaptiveThreshold(long method, long nBoxSize, CxImage* pContrastMask, long nBias, float fGlobalLocalBalance)
{
if (!pDib)
return false;
if (pContrastMask){
if (!pContrastMask->IsValid() ||
!pContrastMask->IsGrayScale() ||
pContrastMask->GetWidth() != GetWidth() ||
pContrastMask->GetHeight() != GetHeight()){
strcpy(info.szLastError,"AdaptiveThreshold invalid ContrastMask");
return false;
}
}
if (nBoxSize<8) nBoxSize = 8;
if (fGlobalLocalBalance<0.0f) fGlobalLocalBalance = 0.0f;
if (fGlobalLocalBalance>1.0f) fGlobalLocalBalance = 1.0f;
long mw = (head.biWidth + nBoxSize - 1)/nBoxSize;
long mh = (head.biHeight + nBoxSize - 1)/nBoxSize;
CxImage mask(mw,mh,8);
if(!mask.GrayScale())
return false;
if(!GrayScale())
return false;
int globalthreshold = OptimalThreshold(method, 0, pContrastMask);
if (globalthreshold <0)
return false;
for (long y=0; y<mh; y++){
for (long x=0; x<mw; x++){
info.nProgress = (long)(100*(x+y*mw)/(mw*mh));
if (info.nEscape) break;
RECT r;
r.left = x*nBoxSize;
r.right = r.left + nBoxSize;
r.bottom = y*nBoxSize;
r.top = r.bottom + nBoxSize;
int threshold = OptimalThreshold(method, &r, pContrastMask);
if (threshold <0) return false;
mask.SetPixelIndex(x,y,(BYTE)max(0,min(255,nBias+((1.0f-fGlobalLocalBalance)*threshold + fGlobalLocalBalance*globalthreshold))));
}
}
mask.Resample(mw*nBoxSize,mh*nBoxSize,0);
mask.Crop(0,head.biHeight,head.biWidth,0);
if(!Threshold(&mask))
return false;
return true;
}
////////////////////////////////////////////////////////////////////////////////
#include <queue>
////////////////////////////////////////////////////////////////////////////////
/**
* Flood Fill
* \param xStart, yStart: starting point
* \param cFillColor: filling color
* \param nTolerance: deviation from the starting point color
* \param nOpacity: can be from 0 (transparent) to 255 (opaque, default)
* \param bSelectFilledArea: if true, the pixels in the region are also set in the selection layer; default = false
* \param nSelectionLevel: if bSelectFilledArea is true, the selected pixels are set to nSelectionLevel; default = 255
* Note: nOpacity=0 && bSelectFilledArea=true act as a "magic wand"
* \return true if everything is ok
*/
bool CxImage::FloodFill(const long xStart, const long yStart, const RGBQUAD cFillColor, const BYTE nTolerance,
BYTE nOpacity, const bool bSelectFilledArea, const BYTE nSelectionLevel)
{
if (!pDib)
return false;
if (!IsInside(xStart,yStart))
return true;
#if CXIMAGE_SUPPORT_SELECTION
if (!SelectionIsInside(xStart,yStart))
return true;
#endif //CXIMAGE_SUPPORT_SELECTION
RGBQUAD* pPalette=NULL;
WORD bpp = GetBpp();
//nTolerance or nOpacity implemented only for grayscale or 24bpp images
if ((nTolerance || nOpacity != 255) && !(head.biBitCount == 24 || IsGrayScale())){
pPalette = new RGBQUAD[head.biClrUsed];
memcpy(pPalette, GetPalette(),GetPaletteSize());
if (!IncreaseBpp(24))
return false;
}
BYTE* pFillMask = (BYTE*)calloc(head.biWidth * head.biHeight,1);
if (!pFillMask)
return false;
//------------------------------------- Begin of Flood Fill
POINT offset[4] = {{-1,0},{0,-1},{1,0},{0,1}};
std::queue<POINT> q;
POINT point = {xStart,yStart};
q.push(point);
if (IsIndexed()){ //--- Generic indexed image, no tolerance OR Grayscale image with tolerance
BYTE idxRef = GetPixelIndex(xStart,yStart);
BYTE idxFill = GetNearestIndex(cFillColor);
BYTE idxMin = (BYTE)min(255, max(0,(int)(idxRef - nTolerance)));
BYTE idxMax = (BYTE)min(255, max(0,(int)(idxRef + nTolerance)));
while(!q.empty())
{
point = q.front();
q.pop();
for (int z=0; z<4; z++){
int x = point.x + offset[z].x;
int y = point.y + offset[z].y;
if(IsInside(x,y)){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
BYTE idx = BlindGetPixelIndex(x, y);
BYTE* pFill = pFillMask + x + y * head.biWidth;
if (*pFill==0 && idxMin <= idx && idx <= idxMax )
{
if (nOpacity>0){
if (nOpacity == 255)
BlindSetPixelIndex(x, y, idxFill);
else
BlindSetPixelIndex(x, y, (BYTE)((idxFill * nOpacity + idx * (255-nOpacity))>>8));
}
POINT pt = {x,y};
q.push(pt);
*pFill = 1;
}
}
}
}
}
} else { //--- RGB image
RGBQUAD cRef = GetPixelColor(xStart,yStart);
RGBQUAD cRefMin, cRefMax;
cRefMin.rgbRed = (BYTE)min(255, max(0,(int)(cRef.rgbRed - nTolerance)));
cRefMin.rgbGreen = (BYTE)min(255, max(0,(int)(cRef.rgbGreen - nTolerance)));
cRefMin.rgbBlue = (BYTE)min(255, max(0,(int)(cRef.rgbBlue - nTolerance)));
cRefMax.rgbRed = (BYTE)min(255, max(0,(int)(cRef.rgbRed + nTolerance)));
cRefMax.rgbGreen = (BYTE)min(255, max(0,(int)(cRef.rgbGreen + nTolerance)));
cRefMax.rgbBlue = (BYTE)min(255, max(0,(int)(cRef.rgbBlue + nTolerance)));
while(!q.empty())
{
point = q.front();
q.pop();
for (int z=0; z<4; z++){
int x = point.x + offset[z].x;
int y = point.y + offset[z].y;
if(IsInside(x,y)){
#if CXIMAGE_SUPPORT_SELECTION
if (BlindSelectionIsInside(x,y))
#endif //CXIMAGE_SUPPORT_SELECTION
{
RGBQUAD cc = BlindGetPixelColor(x, y);
BYTE* pFill = pFillMask + x + y * head.biWidth;
if (*pFill==0 &&
cRefMin.rgbRed <= cc.rgbRed && cc.rgbRed <= cRefMax.rgbRed &&
cRefMin.rgbGreen <= cc.rgbGreen && cc.rgbGreen <= cRefMax.rgbGreen &&
cRefMin.rgbBlue <= cc.rgbBlue && cc.rgbBlue <= cRefMax.rgbBlue )
{
if (nOpacity>0){
if (nOpacity == 255)
BlindSetPixelColor(x, y, cFillColor);
else
{
cc.rgbRed = (BYTE)((cFillColor.rgbRed * nOpacity + cc.rgbRed * (255-nOpacity))>>8);
cc.rgbGreen = (BYTE)((cFillColor.rgbGreen * nOpacity + cc.rgbGreen * (255-nOpacity))>>8);
cc.rgbBlue = (BYTE)((cFillColor.rgbBlue * nOpacity + cc.rgbBlue * (255-nOpacity))>>8);
BlindSetPixelColor(x, y, cc);
}
}
POINT pt = {x,y};
q.push(pt);
*pFill = 1;
}
}
}
}
}
}
if (pFillMask[xStart+yStart*head.biWidth] == 0 && nOpacity>0){
if (nOpacity == 255)
BlindSetPixelColor(xStart, yStart, cFillColor);
else
{
RGBQUAD cc = BlindGetPixelColor(xStart, yStart);
cc.rgbRed = (BYTE)((cFillColor.rgbRed * nOpacity + cc.rgbRed * (255-nOpacity))>>8);
cc.rgbGreen = (BYTE)((cFillColor.rgbGreen * nOpacity + cc.rgbGreen * (255-nOpacity))>>8);
cc.rgbBlue = (BYTE)((cFillColor.rgbBlue * nOpacity + cc.rgbBlue * (255-nOpacity))>>8);
BlindSetPixelColor(xStart, yStart, cc);
}
}
pFillMask[xStart+yStart*head.biWidth] = 1;
//------------------------------------- End of Flood Fill
//if necessary, restore the original BPP and palette
if (pPalette){
DecreaseBpp(bpp, false, pPalette);
delete [] pPalette;
}
#if CXIMAGE_SUPPORT_SELECTION
if (bSelectFilledArea){
if (!SelectionIsValid()){
if (!SelectionCreate()){
return false;
}
SelectionClear();
info.rSelectionBox.right = head.biWidth;
info.rSelectionBox.top = head.biHeight;
info.rSelectionBox.left = info.rSelectionBox.bottom = 0;
}
RECT r;
SelectionGetBox(r);
for (long y = r.bottom; y < r.top; y++){
BYTE* pFill = pFillMask + r.left + y * head.biWidth;
for (long x = r.left; x<r.right; x++){
if (*pFill) SelectionSet(x,y,nSelectionLevel);
pFill++;
}
}
SelectionRebuildBox();
}
#endif //CXIMAGE_SUPPORT_SELECTION
free(pFillMask);
return true;
}
////////////////////////////////////////////////////////////////////////////////
#endif //CXIMAGE_SUPPORT_DSP